srishti-hf1110 commited on
Commit
c67444b
1 Parent(s): 4ccfdd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ Original file is located at
7
  https://colab.research.google.com/drive/1XX8pCT291obpzL4fc1vu5L_HTG027lle
8
  """
9
  import gradio as gr
 
10
  import datasets
11
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
12
 
 
7
  https://colab.research.google.com/drive/1XX8pCT291obpzL4fc1vu5L_HTG027lle
8
  """
9
  import gradio as gr
10
+ import torch
11
  import datasets
12
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
13