Update app.py

app.py

@@ -19,16 +19,19 @@ def download_weights():
     
     for j in range(len(url)):
       if not os.path.isfile(fname[j]):
-        try:
-          r = requests.get(url[j])
-        except requests.ConnectionError:
-          print("!!! Failed to download data !!!")
-        else:
-          if r.status_code != requests.codes.ok:
-            print("!!! Failed to download data !!!")
-          else:
-            with open(fname[j], "wb") as fid:
-              fid.write(r.content)
+        ntries = 0
+        while ntries<10:
+            try:
+              r = requests.get(url[j])
+            except:
+                print("!!! Failed to download data !!!")
+                ntries += 1 
+            
+      if r.status_code != requests.codes.ok:
+        print("!!! Failed to download data !!!")
+      else:
+        with open(fname[j], "wb") as fid:
+          fid.write(r.content)
 
 try:
     download_weights()