Spaces:
Sleeping
Sleeping
app14
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ from pydantic import BaseModel
|
|
| 31 |
import shutil
|
| 32 |
|
| 33 |
# Cell 1: Image Classification Model
|
| 34 |
-
image_pipeline = pipeline(task="image-classification", model="
|
| 35 |
|
| 36 |
def predict_image(input_img):
|
| 37 |
predictions = image_pipeline(input_img)
|
|
|
|
| 31 |
import shutil
|
| 32 |
|
| 33 |
# Cell 1: Image Classification Model
|
| 34 |
+
image_pipeline = pipeline(task="image-classification", model="microsoft/resnet-50")
|
| 35 |
|
| 36 |
def predict_image(input_img):
|
| 37 |
predictions = image_pipeline(input_img)
|