Update app.py

app.py

@@ -4,7 +4,7 @@ from tensorflow.keras.utils import CustomObjectScope
 from tensorflow.keras.layers.experimental.preprocessing import RandomHeight
 
 with CustomObjectScope({'RandomHeight': RandomHeight}):
-    model_0 = tf.keras.models.load_model('bestmodel.h5')
+    model_0 = tf.keras.models.load_model('/content/drive/MyDrive/bestmodel_porno_final_meilleure100%2.0.h5')
 
 def classify_image(inp):
     inp = inp.reshape((-1, 224, 224, 3))
@@ -27,7 +27,7 @@ def classify_image(inp):
     return output
 
 
-image = gr.inputs.Image(shape=(224, 224))
+image = gr.Image(height=224, width=224)
 
 gr.Interface(
     fn=classify_image, inputs=image, outputs="text",live=True, theme="dark-peach",title="API de détection des images violentes",