vishalkatheriya18 commited on
Commit
3500d9f
1 Parent(s): dacd259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,9 +40,9 @@ if url:
40
  st.image(image.resize((200, 200)), caption="Uploaded Image", use_column_width=False)
41
 
42
  start_time = time.time()
43
-
44
  # Predict using YOLO
45
- predictions = predict_with_yolo(image)
46
 
47
  # Display predictions
48
  if predictions:
 
40
  st.image(image.resize((200, 200)), caption="Uploaded Image", use_column_width=False)
41
 
42
  start_time = time.time()
43
+ image_resized = image.resize((224, 224))
44
  # Predict using YOLO
45
+ predictions = predict_with_yolo(image_resized)
46
 
47
  # Display predictions
48
  if predictions: