Spaces:
Running
on
Zero
Running
on
Zero
updated
Browse files
app.py
CHANGED
@@ -16,8 +16,8 @@ pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_
|
|
16 |
@spaces.GPU
|
17 |
def generate(prompt):
|
18 |
response = pipe(prompt, max_new_tokens=512)
|
19 |
-
r = response[0]['generated_text'][-1]['content']
|
20 |
-
return
|
21 |
|
22 |
gr.Interface(
|
23 |
fn=generate,
|
|
|
16 |
@spaces.GPU
|
17 |
def generate(prompt):
|
18 |
response = pipe(prompt, max_new_tokens=512)
|
19 |
+
# r = response[0]['generated_text'][-1]['content']
|
20 |
+
return response
|
21 |
|
22 |
gr.Interface(
|
23 |
fn=generate,
|