roshnn24 commited on
Commit
b936fed
·
verified ·
1 Parent(s): 7ca59a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -153,7 +153,7 @@ def initialize_llm():
153
  llm = HuggingFaceHub(
154
  repo_id="mistralai/Mistral-7B-Instruct-v0.3", # Using a simpler model for testing
155
  huggingfacehub_api_token=api_token,
156
- )
157
 
158
  # Quick test
159
  test = llm("Say hello")
 
153
  llm = HuggingFaceHub(
154
  repo_id="mistralai/Mistral-7B-Instruct-v0.3", # Using a simpler model for testing
155
  huggingfacehub_api_token=api_token,
156
+ task="text-generation" )
157
 
158
  # Quick test
159
  test = llm("Say hello")