jaafarhh commited on
Commit
2ed997d
·
verified ·
1 Parent(s): 09618ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -65,11 +65,11 @@ class DarijaTherapist:
65
  self.llm = HuggingFaceEndpoint(
66
  endpoint_url="https://api-inference.huggingface.co/models/MBZUAI-Paris/Atlas-Chat-9B",
67
  task="text-generation",
 
 
 
68
  model_kwargs={
69
- "temperature": 0.7,
70
  "max_length": 512,
71
- "do_sample": True,
72
- "return_full_text": False
73
  },
74
  huggingfacehub_api_token=os.getenv("HUGGINGFACE_API_TOKEN")
75
  )
 
65
  self.llm = HuggingFaceEndpoint(
66
  endpoint_url="https://api-inference.huggingface.co/models/MBZUAI-Paris/Atlas-Chat-9B",
67
  task="text-generation",
68
+ temperature=0.7,
69
+ do_sample=True,
70
+ return_full_text=False,
71
  model_kwargs={
 
72
  "max_length": 512,
 
 
73
  },
74
  huggingfacehub_api_token=os.getenv("HUGGINGFACE_API_TOKEN")
75
  )