Upload with huggingface_hub

DESCRIPTION.md

@@ -0,0 +1 @@
+This text generation demo takes in input text and returns generated text. It uses the Transformers library to set up the model and has two examples.

README.md

@@ -1,12 +1,11 @@
+
 ---
-title: Text Generation Main
-emoji: 🌖
-colorFrom: purple
-colorTo: gray
+title: text_generation_main 
+emoji: 🔥
+colorFrom: indigo
+colorTo: indigo
 sdk: gradio
 sdk_version: 3.6
-app_file: app.py
+app_file: run.py
 pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

requirements.txt

@@ -0,0 +1,3 @@
+git+https://github.com/huggingface/transformers
+gradio
+torchhttps://gradio-main-build.s3.amazonaws.com/c3bec6153737855510542e8154391f328ac72606/gradio-3.6-py3-none-any.whl

run.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from transformers import pipeline
+
+generator = pipeline('text-generation', model='gpt2')
+
+def generate(text):
+    result = generator(text, max_length=30, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+examples = [
+    ["The Moon's orbit around Earth has"],
+    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
+]
+
+demo = gr.Interface(
+    fn=generate,
+    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+    examples=examples
+)
+
+demo.launch()