Update app.py

app.py

@@ -8,6 +8,10 @@ MODEL_FILE = "./model.onnx"
 session = ort.InferenceSession(MODEL_FILE)
 tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
 
+# Add this after loading the model
+print("Model inputs:", [input.name for input in session.get_inputs()])
+print("Input shapes:", [input.shape for input in session.get_inputs()])
+
 # Gradio prediction function
 def gradio_predict(input_text):
     # Tokenize input text
@@ -41,10 +45,6 @@ def gradio_predict(input_text):
         translated_text = tokenizer.decode(outputs[0][0], skip_special_tokens=True)
         return translated_text
 
-    # Add this after loading the model
-    print("Model inputs:", [input.name for input in session.get_inputs()])
-    print("Input shapes:", [input.shape for input in session.get_inputs()])
-
     except Exception as e:
         print(f"Error during inference: {e}")
         return "An error occurred during inference."
@@ -55,4 +55,4 @@ gr.Interface(
     inputs="text",
     outputs="text",
     live=True
-).launch(share=True)
+).launch()