Update app.py

app.py

@@ -1,46 +1,30 @@
+import os
+import requests
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load TinyLlama (small but smart)
-model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
 
 def generate_notes(topic):
-    topic = topic.strip()
-    if not topic:
+    if not topic.strip():
         return "⚠️ Please enter a topic first."
 
     prompt = (
-        f"You are an expert teacher. Write detailed, well-structured study notes about '{topic}'. "
-        f"Include an introduction, main concepts, key points, and a short summary. "
-        f"Keep the tone educational and clear for students."
+        f"Write detailed, educational notes about '{topic}'. "
+        f"Include clear explanations, key terms, and examples. "
+        f"Use proper paragraphs and avoid repetition."
     )
-
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(
-        **inputs,
-        max_new_tokens=600,
-        temperature=0.7,
-        top_p=0.9,
-        repetition_penalty=1.2,
-    )
-
-    text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return text.replace(prompt, "").strip()
+    payload = {"inputs": prompt, "parameters": {"max_new_tokens": 600, "temperature": 0.7}}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    result = response.json()
+    return result[0]["generated_text"]
 
 iface = gr.Interface(
     fn=generate_notes,
-    inputs=gr.Textbox(lines=2, placeholder="Enter a topic (e.g. Photosynthesis, World Geography)"),
+    inputs=gr.Textbox(lines=2, placeholder="Enter a topic (e.g. Photosynthesis)"),
     outputs=gr.Textbox(lines=15, label="Generated Notes"),
-    title="📚 AI Note Generator (TinyLlama 1.1B Chat)",
-    description="Generate long, educational notes for free using the TinyLlama 1.1B Chat model.",
-    examples=[
-        ["Photosynthesis process"],
-        ["Causes of World War II"],
-        ["Artificial Intelligence"],
-        ["The Water Cycle"]
-    ]
+    title="📗 Detailed Note Generator (Mistral 7B via Hugging Face API)",
+    description="Uses the Mistral-7B model hosted on Hugging Face Inference API (secure token)."
 )
 
 iface.launch()