Spaces:

poltextlab
/

babelmachine

Running

poltextlab commited on Feb 10

Commit

40ba46f

verified ·

1 Parent(s): e493ad0

fix typo

Files changed (1) hide show

interfaces/illframes.py CHANGED Viewed

@@ -68,7 +68,7 @@ def predict_illframes(text, domain):
     return predict(text, model_id, tokenizer_id, label_names)
 demo = gr.Interface(
-    fn=predict_cap,
     inputs=[gr.Textbox(lines=6, label="Input"),
             gr.Dropdown(languages, label="Language"),
             gr.Dropdown(domains.keys(), label="Domain")],

     return predict(text, model_id, tokenizer_id, label_names)
 demo = gr.Interface(
+    fn=predict_illframes,
     inputs=[gr.Textbox(lines=6, label="Input"),
             gr.Dropdown(languages, label="Language"),
             gr.Dropdown(domains.keys(), label="Domain")],