Ajay Singh Rathore commited on
Commit
111c32b
·
1 Parent(s): c108159

Update model to revision v23

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,12 +12,13 @@ logger = logging.getLogger(__name__)
12
  app = FastAPI()
13
 
14
  MODEL_NAME = "Zodex/my-custom-food-model"
 
15
 
16
  # Load the processor and model separately for more control
17
  try:
18
  logger.info(f"Loading model '{MODEL_NAME}'...")
19
  processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
20
- model = AutoModelForImageClassification.from_pretrained(MODEL_NAME)
21
  logger.info("Model and processor loaded successfully!")
22
  except Exception as e:
23
  logger.error(f"Failed to load model: {str(e)}")
 
12
  app = FastAPI()
13
 
14
  MODEL_NAME = "Zodex/my-custom-food-model"
15
+ REVISION_NAME = 'v23'
16
 
17
  # Load the processor and model separately for more control
18
  try:
19
  logger.info(f"Loading model '{MODEL_NAME}'...")
20
  processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
21
+ model = AutoModelForImageClassification.from_pretrained(MODEL_NAME ,revision=REVISION_NAME)
22
  logger.info("Model and processor loaded successfully!")
23
  except Exception as e:
24
  logger.error(f"Failed to load model: {str(e)}")