Update app.py

app.py

@@ -49,7 +49,7 @@ def predict(input_image):
     prediction = loaded_model.predict(input_image)
 
     # A saída será uma matriz de previsões (no caso de classificação de duas classes, será algo como [[probabilidade_classe_0, probabilidade_classe_1]])
-    class_names = ["Not Hot Dog", "Hot Dog"]
+    class_names = ["Normal", "Cataract"]
 
     # Determine a classe mais provável
     predicted_class = class_names[np.argmax(prediction)]
@@ -61,7 +61,7 @@ iface = gr.Interface(
     predict,
     inputs=gr.inputs.Image(label="Upload hot dog candidate", type="file"),
     outputs=gr.outputs.Label(num_top_classes=2),
-    title="Hot Dog? Or Not?",
+    title="Normal? Or Cataract?",
 )
 
 # Execute a interface Gradio