roshnn24 commited on
Commit
e28ec9d
·
verified ·
1 Parent(s): 4f3d8ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -151,9 +151,8 @@ def initialize_llm():
151
 
152
  # Initialize with minimal settings
153
  llm = HuggingFaceHub(
154
- repo_id="microsoft/phi-4", # Using a simpler model for testing
155
  huggingfacehub_api_token=api_token,
156
- task="text-generation"
157
  )
158
 
159
  # Quick test
 
151
 
152
  # Initialize with minimal settings
153
  llm = HuggingFaceHub(
154
+ repo_id="facebook/opt-6.7b", # Using a simpler model for testing
155
  huggingfacehub_api_token=api_token,
 
156
  )
157
 
158
  # Quick test