Update worker.py

worker.py

@@ -30,7 +30,8 @@ def init_llm():
     if not hf_token:
         raise ValueError("HUGGINGFACEHUB_API_TOKEN is not set in environment variables.")
 
-    model_id = "mistralai/Mistral-7B-Instruct-v0.1"
+    model_id = model_id = "tiiuae/falcon-rw-1b"  # Falcon-1B model
+
     hf_pipeline = pipeline("text-generation", model=model_id, device=DEVICE)
     llm_pipeline = HuggingFacePipeline(pipeline=hf_pipeline)