Update app.py
Browse files
app.py
CHANGED
|
@@ -150,9 +150,8 @@ def initialize_llm():
|
|
| 150 |
|
| 151 |
# Initialize with minimal settings
|
| 152 |
llm = HuggingFaceHub(
|
| 153 |
-
repo_id="
|
| 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
|