Update app.py

app.py

@@ -1,3 +1,12 @@
+import gradio as gr
+import onnxruntime as ort
+from transformers import AutoTokenizer
+import numpy as np
+
+MODEL_FILE = "./model.onnx"
+session = ort.InferenceSession(MODEL_FILE)
+tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
+
 def gradio_predict(input_text):
     try:
         # Tokenize input text
@@ -52,3 +61,12 @@ def gradio_predict(input_text):
         import traceback
         print(traceback.format_exc())
         return f"Error during translation: {str(e)}"
+
+# Gradio interface for the web app
+gr.Interface(
+    fn=gradio_predict,
+    inputs="text",
+    outputs="text",
+    live=True
+).launch()
+