Ajay Singh Rathore commited on
Commit ·
111c32b
1
Parent(s): c108159
Update model to revision v23
Browse files
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)}")
|