Spaces:
Sleeping
Sleeping
jovian
commited on
Commit
·
409fbec
1
Parent(s):
20f63c0
two models
Browse files- app.py +2 -2
- model/train_model.pt +3 -0
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from sahi import AutoDetectionModel
|
|
| 6 |
from PIL import Image
|
| 7 |
import plotly.graph_objects as go
|
| 8 |
import torch
|
| 9 |
-
|
| 10 |
|
| 11 |
import os
|
| 12 |
import shutil
|
|
@@ -320,7 +320,7 @@ def upload_image(image):
|
|
| 320 |
"""Process the uploaded image (if needed) and display it."""
|
| 321 |
return image
|
| 322 |
|
| 323 |
-
|
| 324 |
def apply_detection(image):
|
| 325 |
"""Run object detection on the uploaded image and return the annotated image."""
|
| 326 |
# Convert image from PIL to NumPy array
|
|
|
|
| 6 |
from PIL import Image
|
| 7 |
import plotly.graph_objects as go
|
| 8 |
import torch
|
| 9 |
+
import spaces
|
| 10 |
|
| 11 |
import os
|
| 12 |
import shutil
|
|
|
|
| 320 |
"""Process the uploaded image (if needed) and display it."""
|
| 321 |
return image
|
| 322 |
|
| 323 |
+
@spaces.GPU
|
| 324 |
def apply_detection(image):
|
| 325 |
"""Run object detection on the uploaded image and return the annotated image."""
|
| 326 |
# Convert image from PIL to NumPy array
|
model/train_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:60e590c8bd47e06665c576b9ff98343a76f4f1bfc0b065535e10d7a807906489
|
| 3 |
+
size 19174547
|