Update app.py

app.py

@@ -12,49 +12,50 @@ translation_model = AutoModelForSeq2SeqLM.from_pretrained("KarmaCST/nllb-200-dis
 tokenizer = AutoTokenizer.from_pretrained("KarmaCST/nllb-200-distilled-600M-dz-to-en")
 
 
-model = gr.load("models/Purz/face-projection")
-
-
 src_lang="dzo_Tibt"
 tgt_lang="eng_Latn"
 
-def translate_dzongkha_to_english(text):
-    translation_pipeline = pipeline("translation",
-                                model=translation_model,
-                                tokenizer=tokenizer,
-                                src_lang=src_lang,
-                                tgt_lang=tgt_lang)
+# def translate_dzongkha_to_english(text):
+#     translation_pipeline = pipeline("translation",
+#                                 model=translation_model,
+#                                 tokenizer=tokenizer,
+#                                 src_lang=src_lang,
+#                                 tgt_lang=tgt_lang)
     
-    translated_text = translation_pipeline(text)[0]['translation_text']
+#     translated_text = translation_pipeline(text)[0]['translation_text']
 
-    def generate_image(translated_text, seed):
-        if seed is not None:
-            random.seed(seed)
-        
-        if text in [example[0] for example in examples]:
-            print(f"Using example: {text}")
+#     return translated_text
 
-        return model(text)
-    
-    # return model(text)
-
-# model = gr.load("models/Purz/face-projection")
+model = gr.load("models/Purz/face-projection")
 
-# def generate_image(text, seed):
-#     if seed is not None:
-#         random.seed(seed)
+def generate_image(text, seed):
+    translation_pipeline = pipeline("translation",
+                                    model=translation_model,
+                                    tokenizer=tokenizer,
+                                    src_lang=src_lang,
+                                    tgt_lang=tgt_lang)
+    
+    text = translation_pipeline(text)[0]['translation_text']
 
-#     if text in [example[0] for example in examples]:
-#         print(f"Using example: {text}")
+    if seed is not None:
+        random.seed(seed)
 
-#     return model(text)
+    if text in [example[0] for example in examples]:
+        print(f"Using example: {text}")
 
-examples = [
-    ["Humanoid Cat Warrior, Full View", None],
-    ["Warhammer Sisterhood", None],
-    ["Future Robots war", None],
-    ["Fantasy dragon", None]
+    return model(text)
+examples=[
+    ["བྱི་ཅུང་ཚུ་གངས་རི་གི་ཐོག་ཁར་འཕུར།", None],
+    ["པཱ་རོ་ཁྲོམ་གྱི་ཐོག་ཁར་གནམ་གྲུ་འཕུར།",None],
+    ["པཱ་རོ་ཁྲོམ་གྱི་ཐོག་ཁར་ ཤིང་ཚུ་གི་བར་ན་ གནམ་གྲུ་འཕུར་བའི་འཐོང་གནང་།",None],
+    ["སློབ་ཕྲུག་ཚུ་ ཆརཔ་ནང་རྐང་རྩེད་རྩེ་དེས།"]
 ]
+# examples = [
+#     ["Humanoid Cat Warrior, Full View", None],
+#     ["Warhammer Sisterhood", None],
+#     ["Future Robots war", None],
+#     ["Fantasy dragon", None]
+# ]
 
 interface = gr.Interface(
     fn=translate_dzongkha_to_english,