update

.gitignore

@@ -1,3 +1,4 @@
 venv
+flagged
 .headline_detector_model
 .word_vectors_cache

app.py

@@ -8,10 +8,21 @@ model_dict = {
     "fastText": FasttextDetector
 }
 
-def process(input, model, radio_idx):
-    model_used: typing.Union[FasttextDetector, IndoBERTweetDetector, CNNDetector] = model_dict[model]
-    model_used = model_used.load_from_scenario(radio_idx + 1)
-    return model_used.predict_text([input])
+def process(text_input, model_name, radio_idx):
+    model: typing.Union[FasttextDetector, IndoBERTweetDetector, CNNDetector] = model_dict[model_name]
+    model = model.load_from_scenario(radio_idx + 1)
+    print([text_input], model_name, radio_idx)
+    test = [
+        "nama kamu siapa?",
+        "Kapolda Jatim Teddy Minahasa Dikabarkan Ditangkap Terkait Narkoba  https://t.co/LD9X6VFaUR",
+        "RSUD Abepura Papua Kebakaran, Tiga Orang Dilaporkan Tewas"
+    ]
+    results = model.predict_text([text_input])
+    print(results)
+    final_text = "not a headline"
+    if results[0] == 1:
+        final_text = "a headline"
+    return final_text
 
 iface = gr.Interface(
     fn=process,
@@ -25,8 +36,12 @@ iface = gr.Interface(
             list(model_dict.keys()),
             label="Model", 
             info="Model Used",
-            max_choices=1
         ), 
-       gr.Radio([1, 2, 3, 4, 5, 6], type="index", value=1)
+       gr.Radio(
+           [1, 2, 3, 4, 5, 6], 
+           label="Scenario Number",
+           type="index", 
+           value=1, 
+        )
     ], outputs="text")
 iface.launch()

requirements.txt

@@ -1,3 +1,3 @@
-headline-detector
+headline-detector>1.0.2
 gradio
 fastapi>0.95