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