Spaces:

danielsapit
/

JPEG_Artifacts_Removal

Runtime error

danielsapit commited on Mar 22, 2022

Commit

bc5914e

•

1 Parent(s): 7289b18

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,9 +24,9 @@ def inference(input_img, is_gray, input_quality, enable_zoom, zoom, x_shift, y_s
     input_quality = 100 - input_quality
-    #model_pool = 'model_zoo'  # fixed
-    model_pool = '/content/FBCNN/model_zoo'  # fixed
-    model_path = os.path.join(model_pool, model_name)
     if os.path.exists(model_path):
         print(f'loading model from {model_path}')
     else:

     input_quality = 100 - input_quality
+    #model_pool = '/FBCNN/model_zoo'  # fixed
+    #model_path = os.path.join(model_pool, model_name)
+    model_path = model_name
     if os.path.exists(model_path):
         print(f'loading model from {model_path}')
     else: