Update app.py

app.py

@@ -9,8 +9,8 @@ quantization_config = QuantoConfig(
     weights = "int4"
 )
 
-model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1", quantization_config=quantization_config, device_map="auto", token=access_token)
-tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1", token = access_token)
+tokenizer = AutoTokenizer.from_pretrained("ProbeMedicalYonseiMAILab/medllama3-v20")
+model = AutoModelForCausalLM.from_pretrained("ProbeMedicalYonseiMAILab/medllama3-v20")
 
 # Function to generate text using the model
 def generate_text(prompt):
@@ -24,7 +24,7 @@ def generate_text(prompt):
 iface = gr.Interface(
     fn=generate_text,
     inputs=[
-        gr.inputs.Textbox(lines=5, label="Input Prompt"),
+        gr.Textbox(lines=5, label="Input Prompt"),
     ],
     outputs=gr.outputs.Textbox(label="Generated Text"),
     title="MisTRAL Text Generation",