Hugging Face's logo

Spaces:
AnishKumbhar
/
DogDiseasePredictor
Runtime error

App Files Community
2
AnishKumbhar commited on
Commit
1a1929d
•
1 Parent(s): 8da9e7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model = ViTForImageClassification.from_pretrained(
21
  num_labels=num_classes # Specify the number of classes
22
  )
23
 
24
- model = Model()
25
  model.load_state_dict(torch.load('model_state.pth', map_location='cpu'))
26
  # Define a function to preprocess the input image
27
  def preprocess_input(input: fastapi.UploadFile):
 
21
  num_labels=num_classes # Specify the number of classes
22
  )
23
 
24
+
25
  model.load_state_dict(torch.load('model_state.pth', map_location='cpu'))
26
  # Define a function to preprocess the input image
27
  def preprocess_input(input: fastapi.UploadFile):