alexdekan030 commited on
Commit
87a400d
·
verified ·
1 Parent(s): c3ca3c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-awcru-nr8j7")
7
- model = AutoModelForImageClassification.from_pretrained("alexdekan030/autotrain-awcru-nr8j7")
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):