Spaces:
Running
Running
Update app.py
Browse files
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 |
|