Hugging Face's logo

Spaces:
Aanisha
/
Image_to_story
Runtime error

App Files Community
Aanisha commited on
Commit
dbbb340
·
1 Parent(s): 7179322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ description = "Just upload an image, and generate a story for the image! GPT-2 i
12
  title = "Story generator from images using ViT and GPT2"
13
 
14
 
15
- model = VisionEncoderDecoderModel.from_pretrained("gagan3012/ViTGPT2_vizwiz").to('cuda')
16
  vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
17
  tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
18
  story_gpt = AutoModelForCausalLM.from_pretrained("pranavpsv/gpt2-genre-story-generator")
 
12
  title = "Story generator from images using ViT and GPT2"
13
 
14
 
15
+ model = VisionEncoderDecoderModel.from_pretrained("gagan3012/ViTGPT2_vizwiz").to('cpu')
16
  vit_feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224-in21k")
17
  tokenizer = PreTrainedTokenizerFast.from_pretrained("distilgpt2")
18
  story_gpt = AutoModelForCausalLM.from_pretrained("pranavpsv/gpt2-genre-story-generator")