Update app.py

app.py

@@ -18,14 +18,20 @@ import numpy as np
 from PIL import Image
 import gradio as gr
 import torch
+import zipfile
+
 
 torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2017/09/11/15/58/sunset-2739472_1280.jpg', 'sunset.jpg')
 
+torch.hub.download_url_to_file('https://github.com/AsuharietYgvar/AppleNeuralHash2ONNX/files/7008624/apple-neural-hash.zip', '.')
+
+with zipfile.ZipFile('apple-neural-hash.zip', 'r') as zip_ref:
+    zip_ref.extractall('.')
 # Load ONNX model
-session = onnxruntime.InferenceSession('model.onnx')
+session = onnxruntime.InferenceSession('apple-neural-hash/model.onnx')
 
 # Load output hash matrix
-seed1 = open('neuralhash_128x96_seed1.dat', 'rb').read()[128:]
+seed1 = open('apple-neural-hash/neuralhash_128x96_seed1.dat', 'rb').read()[128:]
 seed1 = np.frombuffer(seed1, dtype=np.float32)
 seed1 = seed1.reshape([96, 128])