fixed import error

gpt2_story_gen.py

@@ -1,4 +1,5 @@
 from transformers import pipeline, CLIPProcessor, CLIPModel
+import torch
 
 
 def generate_story(image_caption, image, genre):
@@ -21,7 +22,7 @@ def generate_story(image_caption, image, genre):
         padding=True
         )
     clip_ranker_outputs = clip_ranker_model(**clip_ranker_inputs)
-    logits_per_image = outputs.logits_per_image
+    logits_per_image = clip_ranker_outputs.logits_per_image
     probs = logits_per_image.softmax(dim=1)
     story = stories[torch.argmax(probs).item()]