Update app.py

app.py

@@ -6,7 +6,7 @@ Automatically generated by Colaboratory.
 Original file is located at
     https://colab.research.google.com/drive/1XX8pCT291obpzL4fc1vu5L_HTG027lle
 """
-import gradio
+import gradio as gr
 import datasets
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
@@ -19,7 +19,8 @@ labels = dataset['train'].features['labels'].names
 
 def classify(im):
   features = feature_extractor(im, return_tensors='pt')
-  logits = model(features["pixel_values"])[-1]
+  with torch.no_grad():
+    logits = model(features["pixel_values"])[-1]
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()
   confidences = {label: float(probs[i]) for i, label in enumerate(labels)}