Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Load a smaller model and tokenizer
-model_name = "skkjodhpur/Gemma-Code-Instruct-Finetune-by-skk"  # Consider a smaller model if available
+# Load model and tokenizer
+model_name = "skkjodhpur/Gemma-Code-Instruct-Finetune-by-skk"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -43,7 +43,6 @@ iface = gr.Interface(
     inputs="text", 
     outputs="text",     
     title="Doctors-Patient Chatbot",
-    subtitle="Fine-Tuning GEMMA-2B for Doctor-Patient Interaction",
     description="Ask me any question related to patient concerns. This model is designed for educational and informational purposes only. Please do not use it for medical diagnosis or treatment. Always consult a qualified healthcare provider for medical advice.",
     allow_flagging="never",  # Disable flagging if not needed
 )