Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -187,11 +187,11 @@ def chat():
|
|
| 187 |
repo_id=repo_id,
|
| 188 |
#max_tokens=3000,
|
| 189 |
max_new_tokens=2000,
|
|
|
|
| 190 |
temperature=0.8,
|
| 191 |
huggingfacehub_api_token=HFT,
|
| 192 |
)
|
| 193 |
-
|
| 194 |
-
data= llm(prompt)
|
| 195 |
#data = response.choices[0].message.content
|
| 196 |
|
| 197 |
# filtering the uneccessary context.
|
|
|
|
| 187 |
repo_id=repo_id,
|
| 188 |
#max_tokens=3000,
|
| 189 |
max_new_tokens=2000,
|
| 190 |
+
task = "text-generation",
|
| 191 |
temperature=0.8,
|
| 192 |
huggingfacehub_api_token=HFT,
|
| 193 |
)
|
| 194 |
+
data= llm.invoke(prompt)
|
|
|
|
| 195 |
#data = response.choices[0].message.content
|
| 196 |
|
| 197 |
# filtering the uneccessary context.
|