update app.py and add requirement.txt

app.py

@@ -1,7 +1,30 @@
 import gradio as gr
+import torch
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load the model and tokenizer
+model_name = "maulanayyy/code_translation_codet5"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Function to perform inference
+def translate_code(input_code):
+    # Prepare the input text
+    input_text = f"translate Java to C#: {input_code}"
+    
+    # Tokenize the input
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+    
+    # Generate the output
+    with torch.no_grad():
+        outputs = model.generate(input_ids, max_length=512)
+    
+    # Decode the output
+    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()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+gradio