mrsarthakgupta commited on
Commit
85d6695
1 Parent(s): be18c1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def classify_image(image):
21
  outputs = model(**inputs)
22
 
23
  results = torch.matmul(outputs.pooler_output.squeeze(), matrix.squeeze())
24
- return {'Probability of image being AI generated': torch.nn.functional.sigmoid(results + torch.Tensor([0.9791784]))}
25
 
26
 
27
  # Create Gradio interface
 
21
  outputs = model(**inputs)
22
 
23
  results = torch.matmul(outputs.pooler_output.squeeze(), matrix.squeeze())
24
+ return {'Our confidence of this image being AI generated': torch.nn.functional.sigmoid(results + torch.Tensor([0.9791784]))}
25
 
26
 
27
  # Create Gradio interface