DamarJati commited on
Commit
768d3e2
·
verified ·
1 Parent(s): 50ac885

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ import numpy as np
9
  import onnxruntime as rt
10
  import pandas as pd
11
  from PIL import Image
12
-
13
 
14
 
15
 
@@ -174,7 +174,7 @@ class Predictor:
174
 
175
  return np.expand_dims(image_array, axis=0)
176
 
177
-
178
  def predict(
179
  self,
180
  image,
 
9
  import onnxruntime as rt
10
  import pandas as pd
11
  from PIL import Image
12
+ import spaces
13
 
14
 
15
 
 
174
 
175
  return np.expand_dims(image_array, axis=0)
176
 
177
+ @spaces.GPU()
178
  def predict(
179
  self,
180
  image,