Rausda6 commited on
Commit
5b7678f
·
verified ·
1 Parent(s): b6dc81d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-yh172-uui7d" # 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)
 
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)