Update app.py

app.py

@@ -28,16 +28,16 @@ def predict_and_save(img, generator_model):
     
 def run(image_path):
     model = tf.keras.models.load_model('pretrained')  
+    '''
     img = tf.keras.preprocessing.image.load_img(
         image_path, target_size=(256, 256)
     )
     
-    #https://www.tensorflow.org/api_docs/python/tf/keras/utils/load_img
-    
     img_array = tf.keras.preprocessing.image.img_to_array(img)
     img_array = tf.expand_dims(img_array, 0) 
-    
-    predict_and_save(img_array, model)
+    '''
+    #predict_and_save(img_array, model)
+    predict_and_save(image_path, model)
 
 iface = gr.Interface(run, gr.inputs.Image(shape=(256, 256)), "image")