Spaces:
Running
Running
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 |
|