clean files

app.py

@@ -53,7 +53,7 @@ else:
     os.remove(local_filename)
     print(f"Removed '{local_filename}' after extraction.")
 
-generator = torch.Generator(device="cuda").manual_seed(42)
+generator = torch.Generator(device="cpu").manual_seed(42)
 
 size = (1024, 1024)
 
@@ -182,8 +182,8 @@ with gr.Blocks(css=css) as demo:
             label="Input Image",
             crop_size=(1024, 1024),
             layers=False,
-            height=712,
-            width=712
+            height=1024,
+            width=1024
         )
 
         result = gr.Image(

requirements.txt

@@ -3,10 +3,7 @@ tqdm
 mxnet-mkl==1.6.0
 numpy==1.23.1
 easydict==1.9.0
-scikit-image==0.17.2
-scikit-learn==0.24.2
 opencv-python
-tensorflow
 joblib
 matplotlib
 pandas
@@ -17,5 +14,5 @@ tabulate
 kornia==0.5.0
 webdataset
 packaging
-scikit-learn==0.24.2
 wldhx.yadisk-direct
+torchvision