Upload app.py

app.py

@@ -3,22 +3,29 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load IBM Granite model
-model_id = "ibm/granite-3.3-2b-instruct"
-token = os.getenv("HF_TOKEN")  # Use Hugging Face Secrets
-tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
-model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float32, use_auth_token=token)
+# ✅ Correct IBM Granite model
+model_id = "ibm-granite/granite-3.3-2b-instruct"
+token = os.getenv("HF_TOKEN")  # Load from Hugging Face Secrets
+
+# ✅ Load model and tokenizer using the updated syntax
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=token)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id,
+    token=token,
+    device_map="auto",
+    torch_dtype=torch.float32
+)
 
 def query_granite(prompt):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     outputs = model.generate(**inputs, max_new_tokens=100)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# Page functions
+# Pages
 def home():
     with gr.Blocks() as demo:
         gr.Markdown("# 🏥 Welcome to HealthAI")
-        gr.Markdown("Your intelligent healthcare assistant with real AI support.")
+        gr.Markdown("Your intelligent healthcare assistant using IBM Granite 3.3-2B Instruct.")
     return demo
 
 def symptoms_app():
@@ -69,10 +76,10 @@ def faqs_app():
     with gr.Blocks() as demo:
         gr.Markdown("## ❓ FAQs")
         gr.Markdown("**Q1:** What is HealthAI?")
-        gr.Markdown("**A:** It's an AI assistant to help with health-related queries using IBM Granite model.")
+        gr.Markdown("**A:** It's an AI assistant to help with health-related queries using IBM Granite 3.3-2B.")
     return demo
 
-# Multi-page Gradio app
+# App configuration
 pages = [
     gr.Page(title="🏠 Home", path="/", block=home),
     gr.Page(title="🩺 Symptoms", path="/symptoms", block=symptoms_app),
@@ -83,4 +90,4 @@ pages = [
 ]
 
 app = gr.App(pages=pages)
-app.launch()
+app.launch()