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

Fix: Use correct model name (Zodex/my-indian-food-model-v23)

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,14 +11,14 @@ logger = logging.getLogger(__name__)
11
 
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)}")
 
11
 
12
  app = FastAPI()
13
 
14
+ MODEL_NAME = "Zodex/my-indian-food-model-v23-full"
15
+
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)
22
  logger.info("Model and processor loaded successfully!")
23
  except Exception as e:
24
  logger.error(f"Failed to load model: {str(e)}")