Update app.py

app.py

@@ -8,7 +8,16 @@ import torch.nn.functional as F
 import pandas as pd
 from unzip import unzip
 
-unzip('img.zip')
+zip_filename = 'img.zip'
+target_dir = 'photos'
+
+# Create the target directory if it doesn't exist
+if not os.path.exists(target_dir):
+    os.makedirs(target_dir)
+
+# Open the zip file and extract its contents to the target directory
+with zipfile.ZipFile(zip_filename, 'r') as zip_ref:
+    zip_ref.extractall(target_dir)
 
 device = 'cpu'
 model_path = "weights/ViT-B-32.pt"