Spaces:

AIOmarRehan
/

Animal_Image_Classification_Using_InceptionV3

Sleeping

AIOmarRehan commited on 4 days ago

Commit

892909e

verified ·

1 Parent(s): eb4c49e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from model import predict
 from datasets import load_dataset
 # Load the HF dataset once
-dataset = load_dataset("AIOmarRehan/AnimalsDataset", split="train")
 def classify_image(img: Image.Image):
     label, confidence, probs = predict(img)

 from datasets import load_dataset
 # Load the HF dataset once
+dataset = load_dataset("AIOmarRehan/AnimalsDataset", split="train", streaming=True)
 def classify_image(img: Image.Image):
     label, confidence, probs = predict(img)