Update app.py
Browse files
app.py
CHANGED
@@ -151,7 +151,8 @@ def initialize_llm():
|
|
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 |
)
|
156 |
|
157 |
# Quick test
|
|
|
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
|