khalednabawi11 commited on
Commit
659b0ec
·
verified ·
1 Parent(s): 4e5f5cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -334,11 +334,11 @@ async def chat_fn(query: Query):
334
  # Run blocking inference in thread
335
  loop = asyncio.get_event_loop()
336
  response = await loop.run_in_executor(executor,
337
- lambda: pipe(prompt, max_new_tokens=200,
338
- temperature=0.3,
339
- top_k=50,
340
- do_sample=True,
341
- top_p=0.9)[0]['generated_text'])
342
 
343
  # Parse answer
344
  answer = response.split("Answer:")[-1].strip() if "Answer:" in response else response.split("الإجابة:")[-1].strip()
 
334
  # Run blocking inference in thread
335
  loop = asyncio.get_event_loop()
336
  response = await loop.run_in_executor(executor,
337
+ lambda: pipe(prompt,
338
+ max_new_tokens=512,
339
+ temperature=0.7,
340
+ do_sample = True,
341
+ top_p=0.9)[0]['generated_text']
342
 
343
  # Parse answer
344
  answer = response.split("Answer:")[-1].strip() if "Answer:" in response else response.split("الإجابة:")[-1].strip()