Update app.py

app.py

@@ -21,13 +21,13 @@ enable_gpu = torch.cuda.is_available()
 map_location = torch.device("cuda") if enable_gpu else "cpu"
 from huggingface_hub import hf_hub_download
 from fastai.learner import load_learner
+import tensorflow as tf
 
 model = torch.load("h2z-85epoch.pth",map_location=torch.device('cpu'))
 def generate_img(img_path):
     img = tf.io.read_file(img_path)
     img = tf.image.decode_png(img)
     img = tf.expand_dims(img, axis=0)
-    img = preprocess_test_image(img)
     prediction = model(img, training=False)
     return prediction