Update app.py

app.py

@@ -1,6 +1,12 @@
 from transformers import pipeline, set_seed
+from transformers import MarianMTModel, MarianTokenizer
 import gradio as grad, random, re
 
+# Inisialisasi model terjemahan
+model_name = "Helsinki-NLP/opus-mt-id-en"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
+
 
 gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
 with open("ideas.txt", "r") as f:
@@ -15,6 +21,10 @@ def generate(starting_text):
         starting_text: str = line[random.randrange(0, len(line))].replace("\n", "").lower().capitalize()
         starting_text: str = re.sub(r"[,:\-–.!;?_]", '', starting_text)
 
+    inputs = tokenizer(starting_text, return_tensors="pt", padding=True, truncation=True)
+    outputs = model.generate(**inputs)
+    starting_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
     response = gpt2_pipe(starting_text, max_length=(len(starting_text) + random.randint(60, 90)), num_return_sequences=4)
     response_list = []
     for x in response:
@@ -30,7 +40,7 @@ def generate(starting_text):
         return response_end
 
 
-txt = grad.Textbox(lines=1, label="Initial Text", placeholder="English Text here")
+txt = grad.Textbox(lines=1, label="Initial Text", placeholder="Masukkan Input Bahasa Indonesia")
 out = grad.Textbox(lines=4, label="Generated Prompts")
 
 examples = []