Update app.py

app.py

@@ -1,5 +1,31 @@
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
-from transformers import pipeline
-translator =pipeline("feature-extraction", model="booba-uz/english-uzbek-translation_v1")
-demo=gr.Interface.from_pipeline(translator)
+
+# Load the pre-trained model and tokenizer
+model_ckpt = "booba-uz/english-uzbek-translation_v1"
+tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_ckpt)
+
+# Set source and target languages for the tokenizer
+tokenizer.src_lang = "en"
+tokenizer.tgt_lang = "uz"
+
+# Define the translation function
+def translator(input_text):  
+    inputs = tokenizer(input_text, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=256)
+    translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return translated_text
+
+# Reset function for canceling the input
+def reset_input():
+    return "",  # Reset input text
+
+# Create Gradio interface with Submit and Cancel buttons
+demo = gr.Interface(
+    fn=translator, 
+    inputs="text", 
+    outputs="text",
+)
+
 demo.launch(debug=True)