Update app.py

app.py

@@ -3,10 +3,10 @@ from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
 import os
 
-# Load environment variables
+# Load env variables
 load_dotenv()
 
-# Initialize Hugging Face client
+# Initialize client
 client = InferenceClient(
     provider="hf-inference",
     api_key=os.getenv("HF_API_KEY")
@@ -14,29 +14,27 @@ client = InferenceClient(
 
 app = Flask(__name__)
 
-# Blog generation function
-def generate_blog(paragraph_topic):
+# Blog generator
+def generate_blog(topic):
     try:
-        response = client.chat.completions.create(
-            model="mistralai/Mistral-7B-Instruct-v0.2",
-            messages=[
-                {
-                    "role": "user",
-                    "content": f"Write a well-structured blog paragraph about: {paragraph_topic}"
-                }
-            ],
-            max_tokens=300,
+        prompt = f"Write a well-structured blog paragraph about: {topic}"
+
+        response = client.text_generation(
+            model="google/flan-t5-large",
+            prompt=prompt,
+            max_new_tokens=300,
         )
-        return response.choices[0].message.content
+
+        return response
 
     except Exception as e:
         return f"Error generating blog: {str(e)}"
 
 
-# Route
 @app.route("/", methods=["GET", "POST"])
 def index():
     paragraph = ""
+
     if request.method == "POST":
         topic = request.form.get("topic")
         if topic:
@@ -45,6 +43,5 @@ def index():
     return render_template("index.html", paragraph=paragraph)
 
 
-# Run app
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))