Hugging Face's logo

Spaces:
envision-ai
/
Custom-Object-Detection
Runtime error

App Files Community
Ubuntu commited on
Commit
4ba1c87
·
1 Parent(s): e30329e

a

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -104,7 +104,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
104
  # preds=str(outputs.logits.softmax(1).argmax(1))
105
  # labels=str(data['labels'])
106
 
107
- return preds, preds, preds
108
 
109
  else:
110
  capture = cv2.VideoCapture(video)
 
104
  # preds=str(outputs.logits.softmax(1).argmax(1))
105
  # labels=str(data['labels'])
106
 
107
+ return outputs, outputs, preds
108
 
109
  else:
110
  capture = cv2.VideoCapture(video)
app.py CHANGED
@@ -104,7 +104,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
104
  # preds=str(outputs.logits.softmax(1).argmax(1))
105
  # labels=str(data['labels'])
106
 
107
- return preds, preds, preds
108
 
109
  else:
110
  capture = cv2.VideoCapture(video)
 
104
  # preds=str(outputs.logits.softmax(1).argmax(1))
105
  # labels=str(data['labels'])
106
 
107
+ return outputs, outputs, preds
108
 
109
  else:
110
  capture = cv2.VideoCapture(video)