removed unused imports and added more details

app.py

@@ -19,9 +19,15 @@ def main(pil_image, genre, model="Conceptual", use_beam_search=True):
 
 
 if __name__ == "__main__":
+    title = "Image to Story"
+    article = "Combines the power of [clip prefix captioning](https://github.com/rmokady/CLIP_prefix_caption) with [gpt2 story generator](https://huggingface.co/pranavpsv/genre-story-generator-v2) to create stories of different genres from image"
+    description = "Drop an image and generate stories of different genre based on that image"
+
     interface = gr.Interface(
         main,
-        title="image2story",
+        title=title,
+        description=description,
+        article=article,
         inputs=[
             gr.inputs.Image(type="pil", source="upload", label="Input"),
             gr.inputs.Dropdown(
@@ -38,6 +44,7 @@ if __name__ == "__main__":
             ),
         ],
         outputs=gr.outputs.Textbox(label="Generated story"),
+        examples=[["dog_image.jpg", "action"]],
         enable_queue=True,
     )
     interface.launch()

prefix_clip.py

@@ -1,23 +1,15 @@
 import clip
-import os
 from torch import nn
 import numpy as np
 import torch
 import torch.nn.functional as nnf
-import sys
 import gdown
 from typing import Tuple, List, Union, Optional
 from transformers import (
     GPT2Tokenizer,
     GPT2LMHeadModel,
-    AdamW,
-    get_linear_schedule_with_warmup,
 )
-from tqdm import tqdm, trange
-from google.colab import files
-import skimage.io as io
-import PIL.Image
-from IPython.display import Image
+from tqdm import trange
 
 
 N = type(None)