add requirements

app.py

@@ -1,15 +1,12 @@
-import torch
-from torch import autocast
-from diffusers import DiffusionPipeline
+import gradio as gr
 
-pipe = DiffusionPipeline.from_pretrained("uf/braBeautifulRealistic_brav5",torch_dtype=torch.float16)
+description = "Story generation with GPT-2"
+title = "Generate your own story"
+examples = [["Adventurer is approached by a mysterious stranger in the tavern for a new quest."]]
 
-prompt = "an asian woman with long hair and a black top is posing for a picture with her hand on her chin, 1girl, solo, long_hair, looking_at_viewer, brown_hair, simple_background, black_hair, jewelry, earrings, necklace, lips, black_shirt, ring, realistic"
-with autocast("cuda"):
-    image = pipe(prompt).images[0]
+interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
+            description=description,
+            examples=examples
+)
 
-#图片保存
-image.save(f"astronaut_rides_horse.png")
-
-#或者你可以直接在这个notebook里显示查看. 
-image
+interface.launch()