Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-# Load the model and tokenizer with authentication token if needed
+# Load the model and tokenizer with trust_remote_code=True
 model_name = "Flmc/DISC-MedLLM"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)
 
 # Function to generate responses
 def generate_response(input_text):