Update app.py

app.py

@@ -31,7 +31,15 @@ def inference(input_img, is_gray, input_quality, enable_zoom, zoom, x_shift, y_s
 
     input_quality = 100 - input_quality
 
-    model_path = model_name   
+    model_path = model_name
+
+    if os.path.exists(model_path):
+        print(f'loading model from {model_path}')
+    else:
+        os.makedirs(os.path.dirname(model_path), exist_ok=True)
+        url = 'https://github.com/jiaxi-jiang/FBCNN/releases/download/v1.0/{}'.format(os.path.basename(model_path))
+        r = requests.get(url, allow_redirects=True)
+        open(model_path, 'wb').write(r.content)
 
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')