Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 # Load the model and tokenizer
-model_name = "maulanayyy/code_translation_codet5"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+model_name = "maulanayyy/code_translation_codet5"  # Ganti dengan nama model yang benar
+tokenizer = T5Tokenizer.from_pretrained(model_name)
+model = T5ForConditionalGeneration.from_pretrained(model_name).to("cuda")
 
 # Function to perform inference
 def translate_code(input_code):
@@ -13,7 +13,7 @@ def translate_code(input_code):
     input_text = f"translate Java to C#: {input_code}"
     
     # Tokenize the input
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")  # Pastikan input_ids ada di GPU
     
     # Generate the output
     with torch.no_grad():