Update app.py

app.py

@@ -21,7 +21,7 @@ def classify_image(image):
         outputs = model(**inputs)
     
     results = torch.matmul(outputs.pooler_output.squeeze(), matrix.squeeze())
-    return {'Probability of image being AI generated': torch.nn.functional.sigmoid(results + torch.Tensor([0.9791784]))}
+    return {'Our confidence of this image being AI generated': torch.nn.functional.sigmoid(results + torch.Tensor([0.9791784]))}
 
 
 # Create Gradio interface