roshnn24 commited on
Commit
add1791
·
verified ·
1 Parent(s): 0425cca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -150,9 +150,8 @@ def initialize_llm():
150
 
151
  # Initialize with minimal settings
152
  llm = HuggingFaceHub(
153
- repo_id="mistralai/Mistral-7B-Instruct-v0.3", # Using a simpler model for testing
154
  huggingfacehub_api_token=api_token,
155
- task="text-generation"
156
  )
157
 
158
  # Quick test
 
150
 
151
  # Initialize with minimal settings
152
  llm = HuggingFaceHub(
153
+ repo_id="meta-llama/Llama-3.1-8B-Instruct", # Using a simpler model for testing
154
  huggingfacehub_api_token=api_token,
 
155
  )
156
 
157
  # Quick test