Hugging Face's logo

Spaces:
randomarnab
/
Img_caption_project_using_ViT_GPT2
Runtime error

App Files Community
randomarnab commited on
Commit
461429e
·
1 Parent(s): c7a62b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ Automatically generated by Colaboratory.
6
  Original file is located at
7
  https://colab.research.google.com/drive/1Uvn7yZCyrMpOYNPb7K0G45tQZJVx8LyX
8
  """
 
9
 
10
  from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
11
  import gradio as gr
 
6
  Original file is located at
7
  https://colab.research.google.com/drive/1Uvn7yZCyrMpOYNPb7K0G45tQZJVx8LyX
8
  """
9
+ !pip install torch
10
 
11
  from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
12
  import gradio as gr