Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def chat(user_input):
|
|
| 35 |
return "Îmi pare rău, nu am suficiente date despre acest subiect pentru a răspunde corect."
|
| 36 |
|
| 37 |
prompt = f"User: {user_input}\nAI:"
|
| 38 |
-
response = pipe(prompt, max_new_tokens=
|
| 39 |
return response[len(prompt):].strip()
|
| 40 |
|
| 41 |
iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="Llama3 8B Chat")
|
|
|
|
| 35 |
return "Îmi pare rău, nu am suficiente date despre acest subiect pentru a răspunde corect."
|
| 36 |
|
| 37 |
prompt = f"User: {user_input}\nAI:"
|
| 38 |
+
response = pipe(prompt, max_new_tokens=64, do_sample=True, temperature=0.1)[0]["generated_text"]
|
| 39 |
return response[len(prompt):].strip()
|
| 40 |
|
| 41 |
iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="Llama3 8B Chat")
|