Upload folder using huggingface_hub

app.py

@@ -13,7 +13,7 @@ from pydantic import BaseModel, Field
 try:
     from demo.object_detection.inference import YOLOv10
 except (ImportError, ModuleNotFoundError):
-    from .inference import YOLOv10
+    from inference import YOLOv10
 
 
 cur_dir = Path(__file__).parent
@@ -74,7 +74,7 @@ if __name__ == "__main__":
     import os
 
     if (mode := os.getenv("MODE")) == "UI":
-        stream.ui.launch(server_port=7860, server_name="0.0.0.0")
+        stream.ui.launch(server_port=7860)
     elif mode == "PHONE":
         stream.fastphone(host="0.0.0.0", port=7860)
     else:

index.html

@@ -119,8 +119,6 @@
             top: 20px;
             left: 50%;
             transform: translateX(-50%);
-            background-color: #f44336;
-            color: white;
             padding: 16px 24px;
             border-radius: 4px;
             font-size: 14px;
@@ -128,6 +126,16 @@
             display: none;
             box-shadow: 0 2px 5px rgba(0, 0, 0, 0.2);
         }
+
+        .toast.error {
+            background-color: #f44336;
+            color: white;
+        }
+
+        .toast.warning {
+            background-color: #ffd700;
+            color: black;
+        }
     </style>
 </head>
 
@@ -181,6 +189,7 @@
         function showError(message) {
             const toast = document.getElementById('error-toast');
             toast.textContent = message;
+            toast.className = 'toast error';
             toast.style.display = 'block';
 
             // Hide toast after 5 seconds
@@ -193,6 +202,18 @@
             const config = __RTC_CONFIGURATION__;
             peerConnection = new RTCPeerConnection(config);
 
+            const timeoutId = setTimeout(() => {
+                const toast = document.getElementById('error-toast');
+                toast.textContent = "Connection is taking longer than usual. Are you on a VPN?";
+                toast.className = 'toast warning';
+                toast.style.display = 'block';
+
+                // Hide warning after 5 seconds
+                setTimeout(() => {
+                    toast.style.display = 'none';
+                }, 5000);
+            }, 5000);
+
             try {
                 const stream = await navigator.mediaDevices.getUserMedia({
                     video: true
@@ -263,7 +284,16 @@
                 // Send initial confidence threshold
                 updateConfThreshold(confThreshold.value);
 
+                peerConnection.addEventListener('connectionstatechange', () => {
+                    if (peerConnection.connectionState === 'connected') {
+                        clearTimeout(timeoutId);
+                        const toast = document.getElementById('error-toast');
+                        toast.style.display = 'none';
+                    }
+                });
+
             } catch (err) {
+                clearTimeout(timeoutId);
                 console.error('Error setting up WebRTC:', err);
                 showError('Failed to establish connection. Please try again.');
                 stop();