Kaushik066 commited on
Commit
be45b57
·
verified ·
1 Parent(s): b7d97a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -198,7 +198,7 @@ if picture is not None:
198
  #st.write('Image saved as:',webcam_path)
199
 
200
  ## Create DataLoader for Webcam Image
201
- webcam_ds = dataset_prod_obj.create_dataset(webcam_path, webcam=True)
202
  webcam_dl = DataLoader(webcam_ds, batch_size=BATCH_SIZE)
203
 
204
  ## Run the predictions
 
198
  #st.write('Image saved as:',webcam_path)
199
 
200
  ## Create DataLoader for Webcam Image
201
+ webcam_ds = dataset_prod_obj.create_dataset(picture, webcam=True)
202
  webcam_dl = DataLoader(webcam_ds, batch_size=BATCH_SIZE)
203
 
204
  ## Run the predictions