Update app.py

app.py

@@ -18,6 +18,7 @@ from PIL import Image
 import torch
 import torch.nn.functional as F
 import time
+import gradio as gr
  
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
  
@@ -33,6 +34,5 @@ def predict(image):
     label = model.config.id2label[predicted_class_idx].split(",")[0]
     time.sleep(2)
     return {label: float(predicted_class_prob)}
-import gradio as gr
  
 gr.Interface(predict, gr.Image(type="pil"), "label").launch()