Emilianohack6950 commited on
Commit
b839ca2
1 Parent(s): 59c6c2a

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ generator = pipeline('text-generation', model='Emilianohack6950/M.A.R.I.A')
5
+
6
+ def generate_text(prompt, tokens_to_generate):
7
+ text = generator(prompt, max_length=tokens_to_generate)
8
+ return text[0]['generated_text']
9
+
10
+ gr.Interface(
11
+ generate_text,
12
+ inputs=["text", "number"],
13
+ outputs="text",
14
+ title="Generador de texto GPT-2",
15
+ description="Genera texto basado en un prompt dado utilizando el modelo pre-entrenado de GPT-2.",
16
+ ).launch()