Update app.py

app.py

@@ -22,6 +22,18 @@ def extract_coordinates(text):
         return "No coordinates found"
 
 
+def norm_coordinates(text):
+
+    generator = pipeline("text2text-generation", model="GEODE/mt5-small-coords-norm")
+    
+    # Example input text
+    input_text = "extract_coordinates: " + text
+
+    # Generate prediction using the pipeline
+    predicted_coordinates_from_pipeline = generator(input_text, max_length=128)
+
+    return predicted_coordinates_from_pipeline[0]['generated_text']
+
 examples = [
     "* AACH ou ACH, s. f. petite ville d'Allemagne dans le cercle de Souabe, près de la source de l'Aach. Long. 26. 57. lat. 47. 55.",
     "* ARCALU (Principauté d') petit état des Tartares-Monguls, sur la riviere d'Hoamko, où commence la grande muraille de la Chine, sous le 122e degré de longitude & le 42e de latitude septentrionale.",
@@ -33,7 +45,7 @@ examples = [
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Geographic Coordinate Extractor (CamemBERT NER Demo)")
+    gr.Markdown("## Geographic Coordinate Extractor")
     
     with gr.Row():
         with gr.Column():
@@ -47,10 +59,12 @@ with gr.Blocks() as demo:
             
     
         with gr.Column():
-            out_text = gr.Textbox(label="Extracted coordinates")
-            #out_map = gr.Map(label="Location on Map")
+            out_text = gr.Textbox(label="Extracted coordinates (fine-tuned CamemBERT NER)")
             run_btn.click(fn=extract_coordinates, inputs=inp, outputs=out_text)
 
+            out_text = gr.Textbox(label="Extracted and normalized coordinates (fine-tuned mT5)")
+            run_btn.click(fn=norm_coordinates, inputs=inp, outputs=out_text)
+
     with gr.Row():
         
         gr.Examples(