Hugging Face's logo

Spaces:
all-things-vits
/
class-attention-map
Runtime error

App Files Community
sayakpaul HF Staff commited on
Commit
2a6b8e9
·
1 Parent(s): 045d37d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import matplotlib.pyplot as plt
3
  import numpy as np
4
  import timm
 
5
  from timm import create_model
6
  from timm.models.layers import PatchEmbed
7
  from torchvision.models.feature_extraction import create_feature_extractor
 
2
  import matplotlib.pyplot as plt
3
  import numpy as np
4
  import timm
5
+ import torch
6
  from timm import create_model
7
  from timm.models.layers import PatchEmbed
8
  from torchvision.models.feature_extraction import create_feature_extractor