Update app.py

app.py

@@ -37,8 +37,6 @@ def convert_img_to_tensor(img):
     convert_img = convert_img.repeat(3, 1, 1)
     return convert_img
 
-
-
 def predict(img) -> Dict:
     """Transforms and performs a prediction on img and returns prediction.
     """
@@ -55,8 +53,6 @@ def predict(img) -> Dict:
     # Create a prediction label and prediction probability dictionary for each prediction class (this is the required format for Gradio's output parameter)
     pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
 
-    # Calculate the prediction time
-
     # Return the prediction dictionary and prediction time
     return pred_labels_and_probs