Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
)
|