Update app.py
Browse files
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")
|