Update app.py

app.py

@@ -3,6 +3,8 @@ from huggingface_hub import hf_hub_download
 import regex
 import gradio as gr
 import os
+import asyncio
+import atexit
 
 # Constants
 MAX_INPUT_LENGTH = 10000  # Maximum characters allowed
@@ -70,6 +72,19 @@ def predict_language(text, top_k=3, threshold=0.5):
     
     return "\n\n".join(results)
 
+# Cleanup function to prevent async errors on shutdown
+def cleanup():
+    try:
+        loop = asyncio.get_event_loop()
+        if loop.is_running():
+            loop.stop()
+        if not loop.is_closed():
+            loop.close()
+    except Exception:
+        pass
+
+atexit.register(cleanup)
+
 # Create Gradio interface
 with gr.Blocks(title="OpenLID-v3 Language Identification") as demo:
     # Use HTML with target="_blank" to open in new tab
@@ -113,7 +128,7 @@ with gr.Blocks(title="OpenLID-v3 Language Identification") as demo:
     gr.Examples(
         examples=[
             ["Asebter-a yura s wudem awurman d amagrad s tutlayt taqbaylit."],
-            ["L’interès es d’utilizar un sistèma liure, personalizable e en occitan."],
+            ["L'interès es d'utilizar un sistèma liure, personalizable e en occitan."],
             ["Maskinsjefen er oppteken av å løfta fram dei maritime utdanningane."],
             ["The quick brown fox jumps over the lazy dog."],
             ["Le renard brun rapide saute par-dessus le chien paresseux."],
@@ -148,4 +163,18 @@ with gr.Blocks(title="OpenLID-v3 Language Identification") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # Get port from environment (HF Spaces sets this)
+    port = int(os.environ.get("PORT", 7860))
+    
+    try:
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=port,
+            ssr_mode=False,  # Disable experimental SSR to prevent the error
+            share=False,
+            show_error=True
+        )
+    except KeyboardInterrupt:
+        print("\nShutting down gracefully...")
+    finally:
+        cleanup()