Update app.py

app.py

@@ -12,8 +12,11 @@ def translate_code(input_code):
     # Prepare the input text
     input_text = f"translate Java to C#: {input_code}"
     
+    # Check if GPU is available
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    
     # Tokenize the input
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")  # Pastikan input_ids ada di GPU
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)  # Pastikan input_ids ada di GPU
     
     # Generate the output
     with torch.no_grad():
@@ -23,8 +26,4 @@ def translate_code(input_code):
     translated_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return translated_code
 
-# Create Gradio interface
-demo = gr.Interface(fn=translate_code, inputs="text", outputs="text", title="Java to C# Code Translator", description="Enter Java code to translate it to C#.")
-
-# Launch the interface
-demo.launch()
+# Create