Update app.py

app.py

@@ -46,13 +46,13 @@ def predict(image):
     preprocessed_image = preprocess(image)
     prediction = model.predict(tf.expand_dims(preprocessed_image, axis=0))[0][0]
     if prediction >= 0.5:
-        return 'dog'
+        return 'noncancer'
     else:
-        return 'cat'
+        return 'cancer'
 
 # Interface Gradio
 inputs = gr.inputs.Image()
 outputs = gr.outputs.Textbox()
 
-app = gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title='Cat or Dog Classifier')
+app = gr.Interface(fn=predict, inputs=inputs, outputs=outputs, title='Breast-Histopathology-Classifier')
 app.launch()