Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,8 @@ def init_llm():
|
|
| 31 |
if not hf_token:
|
| 32 |
raise ValueError("HUGGINGFACEHUB_API_TOKEN is not set in environment variables.")
|
| 33 |
|
| 34 |
-
model_id = "tiiuae/falcon-
|
|
|
|
| 35 |
hf_pipeline = pipeline("text-generation", model=model_id, device=DEVICE)
|
| 36 |
llm_pipeline = HuggingFacePipeline(pipeline=hf_pipeline)
|
| 37 |
|
|
|
|
| 31 |
if not hf_token:
|
| 32 |
raise ValueError("HUGGINGFACEHUB_API_TOKEN is not set in environment variables.")
|
| 33 |
|
| 34 |
+
model_id = "tiiuae/falcon-rw-1b"
|
| 35 |
+
|
| 36 |
hf_pipeline = pipeline("text-generation", model=model_id, device=DEVICE)
|
| 37 |
llm_pipeline = HuggingFacePipeline(pipeline=hf_pipeline)
|
| 38 |
|