Spaces:

Haaribo
/

Bird-classification-QSENN

Sleeping

Haaribo commited on Nov 4, 2024

Commit

f61ab18

verified ·

1 Parent(s): a12dd49

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ def genreate_intepriable_output(input,dataset="CUB2011", arch="resnet50",seed=12
         print("outputclass:",output)
-        data_dir=Path.home()/"tmp/Datasets/CUB200/CUB_200_2011/"
         labels = pd.read_csv(data_dir/"image_class_labels.txt", sep=' ', names=['img_id', 'target'])
         namelist=pd.read_csv(data_dir/"images.txt",sep=' ',names=['img_id','file_name'])
         classlist=pd.read_csv(data_dir/"classes.txt",sep=' ',names=['cl_id','class_name'])

         print("outputclass:",output)
+        data_dir=Path("tmp/Datasets/CUB200/CUB_200_2011/")
         labels = pd.read_csv(data_dir/"image_class_labels.txt", sep=' ', names=['img_id', 'target'])
         namelist=pd.read_csv(data_dir/"images.txt",sep=' ',names=['img_id','file_name'])
         classlist=pd.read_csv(data_dir/"classes.txt",sep=' ',names=['cl_id','class_name'])