Update app.py

app.py

@@ -7,11 +7,13 @@ learn = load_learner('food_classif_model_v1.pkl')
 def predict_image(img):
     pred,los,prob = learn.predict(img)
     top_values, top_indx = torch.topk(prob, 3) # show 3 max values
+    text_out = ''
 
     if (top_values < 0.4).all():
-        return 'I do not know what is that'
+        text_out = 'I do not know what is that'
     else:    
-        return f'{food_indx[top_indx[0]]} = {top_values[0]}\n{food_indx[top_indx[1]]} = {top_values[1]}\n{food_indx[top_indx[2]]} = {top_values[2]}'
+        text_out = f'{food_indx[top_indx[0]]} = {top_values[0]}\n{food_indx[top_indx[1]]} = {top_values[1]}\n{food_indx[top_indx[2]]} = {top_values[2]}'
+    return text_out
 
 
 # UI huggface