willcat commited on
Commit
3b39857
1 Parent(s): 9d16267

add requirements

Browse files
Files changed (1) hide show
  1. app.py +12 -4
app.py CHANGED
@@ -1,5 +1,13 @@
1
- # Example of usage
2
- from transformers import pipeline
3
 
4
- story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
5
- print(story_gen("<BOS> <superhero> Batman"))
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from gradio.mix import Series
3
 
4
+ description = "Generate your own D&D story!"
5
+ title = "French Story Generator using Opus MT and GPT-2"
6
+ translator_fr = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-fr-en")
7
+ story_gen = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator")
8
+ translator_en = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-en-fr")
9
+ examples = [["L'aventurier est approché par un mystérieux étranger, pour une nouvelle quête."]]
10
+
11
+ Series(translator_fr, story_gen, translator_en, description = description,
12
+ title = title,
13
+ examples=examples, inputs = gr.inputs.Textbox(lines = 10)).launch()