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