second commit

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 import torch
-from model import create_effnetb2
+from model import create_effnetb2_model
 from timeit import default_timer as timer
 from typing import Tuple,Dict
 
@@ -15,7 +15,7 @@ def predict(img)-> Tuple[Dict,float]:
     effnetb2.eval()
     with torch.inference_mode():
         pred_probs=torch.softmax(effnetb2(img),dim=1)
-    pred_labels_and_probs={class_names[i]:float(pred_probs[0][i] for i in range(len(class_names)))}
+    pred_labels_and_probs={class_names[i]:float(pred_probs[0][i]) for i in range(len(class_names))}
     pred_time=round(timer()-start_timer,5)
     return pred_labels_and_probs,pred_time
 title="FoodVision"