LPX55 commited on
Commit
edf85c0
·
verified ·
1 Parent(s): 9d94875

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -164,6 +164,10 @@ def predict_image(img, confidence_threshold):
164
  'AI': pred[0],
165
  'Real': pred[1]
166
  }
 
 
 
 
167
  # Combine results
168
  combined_results = {
169
  "SwinV2": label_1,
 
164
  'AI': pred[0],
165
  'Real': pred[1]
166
  }
167
+ except Exception as e:
168
+ label_3 = f"Error: {str(e)}"
169
+
170
+
171
  # Combine results
172
  combined_results = {
173
  "SwinV2": label_1,