Create app.py

app.py

@@ -0,0 +1,70 @@
+import gradio as gr
+from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
+import torch
+from langdetect import detect, LangDetectException
+
+# Load model dan tokenizer
+model_name = "facebook/m2m100_418M"
+try:
+    tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+    model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    model.to(device)
+    model.eval()
+except Exception as e:
+    raise Exception(f"Gagal memuat model: {str(e)}")
+
+# Fungsi terjemahan
+def translate_text(text, source_lang=None):
+    try:
+        # Autodeteksi bahasa jika source_lang tidak diberikan
+        if not source_lang:
+            try:
+                detected_lang = detect(text)
+                if detected_lang not in tokenizer.supported_languages:
+                    return f"Bahasa terdeteksi '{detected_lang}' tidak didukung.", detected_lang
+                source_lang = detected_lang
+            except LangDetectException:
+                return "Gagal mendeteksi bahasa. Harap masukkan kode bahasa sumber.", None
+        else:
+            if source_lang not in tokenizer.supported_languages:
+                return f"Kode bahasa '{source_lang}' tidak didukung.", None
+
+        # Set bahasa sumber
+        tokenizer.src_lang = source_lang
+
+        # Encode dan terjemahkan
+        encoded = tokenizer(text, return_tensors="pt", padding=True, truncation=True).to(device)
+        generated_tokens = model.generate(
+            **encoded,
+            forced_bos_token_id=tokenizer.get_lang_id("en")
+        )
+        translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+
+        return translated_text, source_lang
+
+    except Exception as e:
+        return f"Error: {str(e)}", None
+
+# Buat antarmuka Gradio
+iface = gr.Interface(
+    fn=translate_text,
+    inputs=[
+        gr.Textbox(label="Teks untuk Diterjemahkan"),
+        gr.Dropdown(
+            choices=["id", "fr", "es", "de", "ja", ""],  # Tambahkan lebih banyak kode bahasa jika perlu
+            label="Bahasa Sumber (kosongkan untuk autodeteksi)",
+            value=""
+        )
+    ],
+    outputs=[
+        gr.Textbox(label="Terjemahan ke Bahasa Inggris"),
+        gr.Textbox(label="Bahasa Sumber Terdeteksi")
+    ],
+    title="M2M100 Translation to English",
+    description="Masukkan teks untuk diterjemahkan ke bahasa Inggris. Biarkan bahasa sumber kosong untuk autodeteksi."
+)
+
+# Luncurkan aplikasi
+if __name__ == "__main__":
+    iface.launch()