Update app.py

app.py

@@ -1,15 +1,24 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
+from transformers import pipeline
 
-# Load your fine-tuned CamemBERT NER model
-model_name = "GEODE/camembert-base-edda-span-classification"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForTokenClassification.from_pretrained(model_name)
 
 
-ner_pipeline = pipeline("token-classification", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
+binary_classifier = pipeline("text-classification", model="GEODE/bert-base-multilingual-cased-binary-classifier-edda-coords")
+ner_pipeline = pipeline("token-classification", model="GEODE/camembert-base-edda-span-classification", aggregation_strategy="simple")
+generator = pipeline("text2text-generation", model="GEODE/mt5-small-coords-norm")
+
+def detect_coordinates(text):
+    # Run binary classification
+    result = binary_classifier(text)
+    
+    if result[0]['label'] == 'Positive':
+        return "Coordinates found"
+    else:
+        return "No coordinates found"
 
 def extract_coordinates(text):
+    if detect_coordinates(text) == "No coordinates found":
+        return "No coordinates found"
     # Run NER
     entities = ner_pipeline(text)
     
@@ -21,11 +30,11 @@ def extract_coordinates(text):
     else:
         return "No coordinates found"
 
+# bert-base-multilingual-cased-binary-classifier-edda-coords
 
 def norm_coordinates(text):
-
-    generator = pipeline("text2text-generation", model="GEODE/mt5-small-coords-norm")
-    
+    if detect_coordinates(text) == "No coordinates found":
+        return "No coordinates found"
     # Example input text
     input_text = "extract_coordinates: " + text
 
@@ -60,6 +69,9 @@ with gr.Blocks() as demo:
             
     
         with gr.Column():
+            out_text = gr.Textbox(label="Extracted coordinates (fine-tuned CamemBERT NER)")
+            run_btn.click(fn=detect_coordinates, inputs=inp, outputs=out_text)
+
             out_text = gr.Textbox(label="Extracted coordinates (fine-tuned CamemBERT NER)")
             run_btn.click(fn=extract_coordinates, inputs=inp, outputs=out_text)