Add files via upload

README.md

@@ -1 +1,12 @@
-# MLopsLocal
+---
+title: MLOPS 1.1
+emoji: 📉
+colorFrom: gray
+colorTo: pink
+sdk: gradio
+sdk_version: 4.44.0
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,76 @@
+
+import gradio as gr
+from transformers import pipeline
+
+# Initialize the translation pipeline with T5-base model
+translator = pipeline("translation", model="google-t5/t5-base")
+
+
+# Define available languages
+AVAILABLE_LANGUAGES = ["French", "German", "Romanian"]
+
+def translate(input_text, target_language):
+    target_language = target_language.lower()
+    
+    # Set the task prefix based on the chosen language
+    if target_language == "french":
+        task_prefix = "translate English to French: "
+    elif target_language == "german":
+        task_prefix = "translate English to German: "
+    elif target_language == "romanian":
+        task_prefix = "translate English to Romanian: "
+    else:
+        return "Sorry, this language is not supported."
+
+    # Perform translation
+    full_input = task_prefix + input_text
+    translated = translator(full_input, max_length=128)
+    return translated[0]['translation_text']
+
+def show_available_languages():
+    return ", ".join(AVAILABLE_LANGUAGES)
+
+def run_tests():
+    test_cases = [
+        ("Hello", "French"),
+        ("How are you?", "German"),
+        ("Good morning", "Romanian"),
+        ("Hello", "Spanish")
+    ]
+    
+    results = []
+    for input_text, target_lang in test_cases:
+        try:
+            result = translate(input_text, target_lang)
+            status = "Passed" if result != "Sorry, this language is not supported." else "Failed"
+            results.append(f"Test '{input_text}' to {target_lang}: {status}\nResult: {result}")
+        except Exception as e:
+            results.append(f"Test '{input_text}' to {target_lang}: Failed - {str(e)}")
+    
+    return "\n".join(results)
+
+# Create Gradio interface
+with gr.Blocks() as iface:
+    gr.Markdown("# Speech Translation using T5-base")
+    gr.Markdown("Translate English text to French, German, or Romanian using the T5-base model.")
+    
+    with gr.Row():
+        input_text = gr.Textbox(label="Enter English text", placeholder="Hi, how are you?")
+        output_text = gr.Textbox(label="Translation")
+    
+    target_lang = gr.Dropdown(AVAILABLE_LANGUAGES, label="Target Language")
+    translate_btn = gr.Button("Translate")
+    
+    with gr.Row():
+        show_langs_btn = gr.Button("Show Available Languages")
+        available_langs = gr.Textbox(label="Available Languages")
+    
+    test_btn = gr.Button("Run Tests")
+    test_output = gr.Textbox(label="Test Results")
+    
+    translate_btn.click(translate, inputs=[input_text, target_lang], outputs=output_text)
+    show_langs_btn.click(show_available_languages, inputs=None, outputs=available_langs)
+    test_btn.click(run_tests, inputs=None, outputs=test_output)
+
+# Launch the app
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio==3.50.2
+transformers==4.36.2
+torch==2.1.2
+sentencepiece==0.1.99

test-file.py

@@ -0,0 +1,26 @@
+import pytest
+from app import translate
+
+def test_translation_to_french():
+    print("Starting French translation test")
+    result = translate("Hello", "French")
+    print(f"French translation result: {result}")
+    assert result.lower() in ["bonjour", "salut"], f"Expected 'Bonjour' or 'Salut', but got {result}"
+
+def test_translation_to_german():
+    print("Starting German translation test")
+    result = translate("Hello", "German")
+    print(f"German translation result: {result}")
+    assert result.lower() == "hallo", f"Expected 'Hallo', but got {result}"
+
+def test_translation_to_romanian():
+    print("Starting Romanian translation test")
+    result = translate("Hello", "Romanian")
+    print(f"Romanian translation result: {result}")
+    assert result.lower() == "salut", f"Expected 'Salut', but got {result}"
+
+def test_unsupported_language():
+    print("Starting unsupported language test")
+    result = translate("Hello", "Spanish")
+    print(f"Unsupported language result: {result}")
+    assert result == "Sorry, this language is not supported."