Spaces:
Runtime error
Runtime error
Fx
Browse files
app.py
CHANGED
@@ -17,8 +17,6 @@ def generate_text(prompt):
|
|
17 |
prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
|
18 |
outputs = generator(
|
19 |
prompt,
|
20 |
-
do_sample=True
|
21 |
-
return_full_text=False
|
22 |
)
|
23 |
# Extract the generated text and return it
|
24 |
generated_text = outputs[0]['generated_text']
|
|
|
17 |
prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
|
18 |
outputs = generator(
|
19 |
prompt,
|
|
|
|
|
20 |
)
|
21 |
# Extract the generated text and return it
|
22 |
generated_text = outputs[0]['generated_text']
|