Rausda6 commited on
Commit
63487f7
·
verified ·
1 Parent(s): a4f15a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoImageProcessor, AutoModelForImageClassification
3
  from PIL import Image
4
  import torch
5
  import numpy as np
6
- import space
7
  # Load model and processor from the Hugging Face Hub
8
  MODEL_REPO = "Rausda6/autotrain-yh172-uui7d" # Replace with your actual model repo name
9
  model = AutoModelForImageClassification.from_pretrained(MODEL_REPO)
 
3
  from PIL import Image
4
  import torch
5
  import numpy as np
6
+ import spaces
7
  # Load model and processor from the Hugging Face Hub
8
  MODEL_REPO = "Rausda6/autotrain-yh172-uui7d" # Replace with your actual model repo name
9
  model = AutoModelForImageClassification.from_pretrained(MODEL_REPO)