Update model/model.py
Browse files- model/model.py +1 -1
    	
        model/model.py
    CHANGED
    
    | @@ -5,7 +5,7 @@ import numpy as np | |
| 5 | 
             
            from PIL import Image
         | 
| 6 |  | 
| 7 | 
             
            def predict(image):
         | 
| 8 | 
            -
                #  | 
| 9 | 
             
                model = AutoModelForImageClassification.from_pretrained('sensei-ml/concrete_crack_images_classification')
         | 
| 10 | 
             
                device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         | 
| 11 | 
             
                model.eval()
         | 
|  | |
| 5 | 
             
            from PIL import Image
         | 
| 6 |  | 
| 7 | 
             
            def predict(image):
         | 
| 8 | 
            +
                # Load the model
         | 
| 9 | 
             
                model = AutoModelForImageClassification.from_pretrained('sensei-ml/concrete_crack_images_classification')
         | 
| 10 | 
             
                device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         | 
| 11 | 
             
                model.eval()
         |