Update app.py

app.py

@@ -38,7 +38,7 @@ else:
 
 # Initialize the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
-model = ORTModelForSeq2SeqLM.from_pretrained(model_id, token=hf_token)
+model = MT5ForConditionalGeneration.from_pretrained(model_id, token=hf_token)
 
 # Create the text generation pipeline
 generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)