Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import torch
-from transformers import T5Tokenizer, T5ForConditionalGeneration
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Load the model and tokenizer
 model_name = "maulanayyy/codet5_code_translation"  # Ganti dengan nama model yang benar
-tokenizer = T5Tokenizer.from_pretrained(model_name)
-model = T5ForConditionalGeneration.from_pretrained(model_name).to("cuda")
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 # Function to perform inference
 def translate_code(input_code):