Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
 import torch
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import T5ForConditionalGeneration, RobertaTokenizer
 import re
 import ast
 
 # Load the fine-tuned model and tokenizer
 model_repo_path = 'sabssag/Latex_to_Python_CodeT5-base'
-model = AutoModelForSeq2SeqLM.from_pretrained(model_repo_path)
-tokenizer = AutoTokenizer.from_pretrained(model_repo_path)
+model = T5ForConditionalGeneration.from_pretrained(model_repo_path, torch_dtype=torch.float16)
+tokenizer = RobertaTokenizer.from_pretrained(model_repo_path)
 model.eval()
 
 # Fix unmatched brackets