Update app.py

app.py

@@ -21,13 +21,9 @@ 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)
     prediction = model(img, training=False)
     return prediction