Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def load_embedding_model():
|
|
| 39 |
def load_llm(api_key):
|
| 40 |
"""Loads the LLM model for answering queries."""
|
| 41 |
return HuggingFaceHub(
|
| 42 |
-
repo_id="
|
| 43 |
huggingfacehub_api_token=api_key,
|
| 44 |
model_kwargs={"temperature": 0.5, "max_length": 100}
|
| 45 |
)
|
|
|
|
| 39 |
def load_llm(api_key):
|
| 40 |
"""Loads the LLM model for answering queries."""
|
| 41 |
return HuggingFaceHub(
|
| 42 |
+
repo_id="distributed/optimized-gpt2-1b",
|
| 43 |
huggingfacehub_api_token=api_key,
|
| 44 |
model_kwargs={"temperature": 0.5, "max_length": 100}
|
| 45 |
)
|