Update app.py

app.py

@@ -18,8 +18,8 @@ def predict(image):
     # Get the predicted class
     predicted_class = logits.argmax(-1).item()
     # You may need to adjust the following line based on your class labels
-    class_names = ["class1", "class2", "class3", "class4"]
-    return predicted_class
+    class_names = ["glioma", "meningioma", "notumor", "pituitary"]
+    return class_names[predicted_class]
 
 # Set up the Gradio interface
 image_cp = gr.Image(type="pil", label='Brain')