changed to api iinterfaceeee small version model api

app.py

@@ -7,7 +7,8 @@ import gradio as gr
 token = os.environ.get("OPENAI_API_KEY")
 # generator = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", token=token)
 
-API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
+# API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
+API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-rw-1b"
 headers = {"Authorization": f"Bearer {token}"}
 
 # generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
@@ -49,8 +50,12 @@ def generate_blog(topic):
     Make it informative and conversational.
     """
     output = query({"inputs": prompt})
+    print("API Response:", output)  # <-- Add this for debugging
+
     if isinstance(output, list) and "generated_text" in output[0]:
         return output[0]["generated_text"]
+    elif "error" in output:
+        return f"Error from model: {output['error']}"
     else:
         return "Failed to generate blog. Please try again."