Hugging Face's logo

Spaces:
SatwikKambham
/
Land-Use-Classifier
Runtime error

App Files Community
SatwikKambham commited on
Commit
e6a5948
•
1 Parent(s): 448703c

Specify providers

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -33,7 +33,10 @@ CATEGORIES = [
33
  class Classifier:
34
  def __init__(self, model_path):
35
  self.model_path = model_path
36
- self.session = ort.InferenceSession(self.model_path)
 
 
 
37
 
38
  self.img_transforms = tv.transforms.Compose(
39
  [
 
33
  class Classifier:
34
  def __init__(self, model_path):
35
  self.model_path = model_path
36
+ self.session = ort.InferenceSession(
37
+ self.model_path,
38
+ providers=["AzureExecutionProvider", "CPUExecutionProvider"],
39
+ )
40
 
41
  self.img_transforms = tv.transforms.Compose(
42
  [