p[

app.py

@@ -13,21 +13,21 @@ except Exception as e:
 
 try:
     fine_tuned_model_1 = pipeline("translation", model="perkan/shortS-opus-mt-tc-base-en-sr")
-    print("Fine-tuned model 1 loaded successfully.")
+    print("Fine-tuned model S loaded successfully.")
 except Exception as e:
-    print(f"Error loading fine-tuned model 1: {e}")
+    print(f"Error loading fine-tuned model S: {e}")
 
 try:
     fine_tuned_model_2 = pipeline("translation", model="perkan/shortM-opus-mt-tc-base-en-sr")
-    print("Fine-tuned model 2 loaded successfully.")
+    print("Fine-tuned model M loaded successfully.")
 except Exception as e:
-    print(f"Error loading fine-tuned model 2: {e}")
+    print(f"Error loading fine-tuned model M: {e}")
 
 try:
     fine_tuned_model_3 = pipeline("translation", model="perkan/shortL-opus-mt-tc-base-en-sr")
-    print("Fine-tuned model 3 loaded successfully.")
+    print("Fine-tuned model L loaded successfully.")
 except Exception as e:
-    print(f"Error loading fine-tuned model 3: {e}")
+    print(f"Error loading fine-tuned model L: {e}")
 
 # Define translation functions
 def translate_base(text):
@@ -38,11 +38,11 @@ def translate_base(text):
 
 def translate_fine_tuned(text, model):
     try:
-        if model == 'model1':
+        if model == 'S model':
             return fine_tuned_model_1(text)[0]['translation_text']
-        elif model == 'model2':
+        elif model == 'M model':
             return fine_tuned_model_2(text)[0]['translation_text']
-        elif model == 'model3':
+        elif model == 'L model':
             return fine_tuned_model_3(text)[0]['translation_text']
         else:
             return "Invalid model selected"
@@ -64,7 +64,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             gr.Markdown("### Fine-tuned Models")
             fine_tuned_input = gr.Textbox(placeholder="Enter text to translate", label="Input")
-            model_select = gr.Dropdown(choices=["model1", "model2", "model3"], label="Select Model")
+            model_select = gr.Dropdown(choices=["S model", "M model", "L model"], label="Select Model")
             fine_tuned_output = gr.Textbox(label="Translation")
             fine_tuned_translate_btn = gr.Button("Translate")
             fine_tuned_translate_btn.click(translate_fine_tuned, inputs=[fine_tuned_input, model_select], outputs=fine_tuned_output)