jovian commited on
Commit
409fbec
·
1 Parent(s): 20f63c0

two models

Browse files
Files changed (2) hide show
  1. app.py +2 -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
- #import spaces
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
- #@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
 
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