tournas commited on
Commit
cb70b02
·
verified ·
1 Parent(s): 6c10423

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -38,6 +38,7 @@ summarizer = pipeline(
38
  )
39
  @spaces.GPU
40
  def detect_objects(image):
 
41
  image_array = np.array(image) # Μετατροπή PIL → NumPy
42
  results = yolo_model(image_array)
43
  detected_objects = []
 
38
  )
39
  @spaces.GPU
40
  def detect_objects(image):
41
+ yolo_model.to('cuda')
42
  image_array = np.array(image) # Μετατροπή PIL → NumPy
43
  results = yolo_model(image_array)
44
  detected_objects = []