Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ logging.basicConfig(level=logging.DEBUG)
|
|
11 |
logger = logging.getLogger(__name__)
|
12 |
|
13 |
# Load model and processor from the Hugging Face Hub
|
14 |
-
MODEL_REPO = "Rausda6/autotrain-
|
15 |
logger.debug(f"Loading model from: {MODEL_REPO}")
|
16 |
model = AutoModelForImageClassification.from_pretrained(MODEL_REPO)
|
17 |
processor = AutoImageProcessor.from_pretrained(MODEL_REPO)
|
|
|
11 |
logger = logging.getLogger(__name__)
|
12 |
|
13 |
# Load model and processor from the Hugging Face Hub
|
14 |
+
MODEL_REPO = "Rausda6/autotrain-uo2t1-gvgzu" # Replace with your actual model repo name
|
15 |
logger.debug(f"Loading model from: {MODEL_REPO}")
|
16 |
model = AutoModelForImageClassification.from_pretrained(MODEL_REPO)
|
17 |
processor = AutoImageProcessor.from_pretrained(MODEL_REPO)
|