WebashalarForML commited on
Commit
dc86108
·
verified ·
1 Parent(s): 57dd549

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.