Update app.py

app.py

@@ -4,17 +4,26 @@ from fastai import *
 import torch
 import os
 from PIL import Image
+import pandas as pd
 
 
 
 model_path  = 'multi_target_resnet34.pkl'
+helper_csv  = 'info.csv'
+df =  pd.read_csv(helper_csv)
 
 model = load_learner(model_path)
 
 def result(path):    
   pred,_,probability = model.predict(path)
-  pred = str(pred)
-  return pred
+  arr = ['Species','Scientific Name','Name','Habitat','Color','Found_In','Diet']
+  vals = ['','','','','','','','']
+  for x in pred:
+      val = df[df['columns'] == x]['values']
+      ind = arr.find(val)
+      vals[ind] = x 
+      
+  return f'{arr}-->{vals}'
 
     
 path = 'test images/'