Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 import numpy as np
 from PIL import Image
 import tensorflow.keras as keras
-import keras.applications.vgg16 as vgg16
+#import keras.applications.vgg16 as vgg16
 
 from tensorflow.keras.models import load_model
 
@@ -14,7 +14,7 @@ classnames=["Fire","Non-Fire"]
 def predict_image(img):
     img_4d=img.reshape(-1,180,180,3)
     prediction=model.predict(img_4d)[0]
-    return {classname[i]: float(prediction[i] for i in range(5))}
+    return {classnames[i]: float(prediction[i] for i in range(2))}
 
 image=gr.inputs.Image(shape=(180,180))
 label=gr.outputs.Label(num_top_classes=2)    
@@ -22,4 +22,4 @@ label=gr.outputs.Label(num_top_classes=2)
 artice="<p style='text-align: center'>Made by Mahak xoxo</p>"
 
 
-gr.Interface(fn=predict_image,inputs=image,title="Forest Fire Classifier",description="THis is a forest fire classification model ",outputs=label,article=article,enable_queue=enable_queue,interpretation='default').launch(share=True)
+gr.Interface(fn=predict_image,inputs=image,title="Forest Fire Classifier",description="This is a forest fire classification model ",outputs=label,article=article,enable_queue=True,interpretation='default').launch(share=True)