Update app.py

app.py

@@ -65,13 +65,13 @@ def gradio_generate_text(prompt, max_length=100, num_return_sequences=1, top_p=0
 # Ensure the models directory exists
 if not os.path.exists('models'):
     os.makedirs('models')
-if not os.path.exists('models/vi-medical-t5-finetune-qa'):
+if not os.path.exists('models/vi-medical-mt5-finetune-qa'):
     # Run the Git LFS commands to clone the model
     run_shell_command('git lfs install')
-    run_shell_command('cd models && git clone https://huggingface.co/danhtran2mind/vi-medical-t5-finetune-qa && cd ..')
+    run_shell_command('cd models && git clone https://huggingface.co/danhtran2mind/vi-medical-mt5-finetune-qa && cd ..')
 
 # Load the trained model and tokenizer
-model_path = "models/vi-medical-t5-finetune-qa"
+model_path = "models/vi-medical-mt5-finetune-qa"
 tokenizer, model, device = load_model_and_tokenizer(model_path)
 # Create Gradio interface