eduard76 commited on
Commit
d8f3012
·
verified ·
1 Parent(s): f9ed50d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=200, do_sample=True, temperature=0.7)[0]["generated_text"]
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")