Initialize app

app.py

@@ -0,0 +1,17 @@
+import gradio as gr
+from transformers import pipeline
+
+text_generator = pipeline(task="text-generation")
+
+def generate_text(prompt):
+  output = text_generator(prompt, max_length=100, num_return_sequences=1)
+  return output[0]["generated_text"]
+
+iface = gr.Interface(
+  fn=generate_text,
+  inputs="textbox",
+  outputs="label",
+  title="Text Generator"
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio==4.27.0
+transformers==4.40.0
+torch==2.2.2