Update app.py

app.py

@@ -8,7 +8,14 @@ model = CustomModel()
 def predict(image: Image.Image):
     # Get predictions from the model
     label, confidence = model.predict(image)
-    return f"Predicted label: {label}", f"Confidence: {confidence:.2f}%"
+    
+    # Determine the result based on the label
+    if label == 1:
+        result = "AI image"
+    else:
+        result = "Real image"
+        
+    return result, f"Confidence: {confidence:.2f}%"
 
 # Define the Gradio interface with updated API
 demo = gr.Interface(