nightfury commited on
Commit
817b56c
1 Parent(s): 29e113e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -55,6 +55,7 @@ def predict(input_image:Image.Image, true_label:str):
55
  face = mtcnn(input_image)
56
  if face is None:
57
  raise Exception('No face detected')
 
58
  face = face.unsqueeze(0) # add the batch dimension
59
  face = F.interpolate(face, size=(256, 256), mode='bilinear', align_corners=False)
60
 
 
55
  face = mtcnn(input_image)
56
  if face is None:
57
  raise Exception('No face detected')
58
+ return "No Photoreal face detected"
59
  face = face.unsqueeze(0) # add the batch dimension
60
  face = F.interpolate(face, size=(256, 256), mode='bilinear', align_corners=False)
61