Update app.py

app.py

@@ -3,9 +3,16 @@ import os
 from flask import Flask, request, jsonify
 from transformers import pipeline
 
+
 from transformers import pipeline
+from huggingface_hub import login
+
+# Log in with your Hugging Face token
+login(token="your_huggingface_token")
+
+# Load the model
+llm = pipeline("text-generation", model="tiiuae/falcon-1b-instruct")
 
-llm = pipeline("text-generation", model="tiiuae/falcon-1b-instruct")  # 1B model instead of 7B