Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = AutoTokenizer.from_pretrained("alibidaran/medical_transcription_generator")
 
-model = AutoModelForCausalLM.from_pretrained("alibidaran/medical_transcription_generator",use_safetensors=True).to(device)
+model = AutoModelForCausalLM.from_pretrained("alibidaran/medical_transcription_generator").to(device)
 def generate_text(Text,Max_length,Temperature):
   torch.manual_seed(32)
   tokenizer.pad_token_id=tokenizer.eos_token_id