Deepakraj2006 commited on
Commit
fe41991
·
verified ·
1 Parent(s): 0ccdb83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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-7b-instruct"
 
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