Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ from transformers import pipeline, AutoImageProcessor, AutoModelForImageClassifi
|
|
3 |
import torch
|
4 |
|
5 |
# Initialize the model and processor
|
6 |
-
processor = AutoImageProcessor.from_pretrained("alexdekan030/autotrain-
|
7 |
-
model = AutoModelForImageClassification.from_pretrained("alexdekan030/autotrain-
|
8 |
pipe = pipeline("image-classification", model=model, image_processor=processor)
|
9 |
|
10 |
def predict_pneumonia(image):
|
|
|
3 |
import torch
|
4 |
|
5 |
# Initialize the model and processor
|
6 |
+
processor = AutoImageProcessor.from_pretrained("alexdekan030/autotrain-1c9rx-gdx3r")
|
7 |
+
model = AutoModelForImageClassification.from_pretrained("alexdekan030/autotrain-1c9rx-gdx3r")
|
8 |
pipe = pipeline("image-classification", model=model, image_processor=processor)
|
9 |
|
10 |
def predict_pneumonia(image):
|