jfforero commited on
Commit
017d1d3
·
verified ·
1 Parent(s): c26322a

Upload colab2html_poetry.py

Browse files
Files changed (1) hide show
  1. colab2html_poetry.py +34 -0
colab2html_poetry.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """Colab2HTML-poetry.ipynb
3
+
4
+ Automatically generated by Colab.
5
+
6
+ Original file is located at
7
+ https://colab.research.google.com/drive/1cJqnlR460ib_c9FpeyPhgoTnybyarDy6
8
+ """
9
+
10
+ !pip install transformers
11
+ verbose = 0
12
+
13
+ #@title UI { display-mode: "form" }
14
+ from transformers import pipeline
15
+ generator = pipeline('text-generation',
16
+ model='huggingtweets/arthur_rimbaud')
17
+ #@title UI { display-mode: "form" }
18
+ text = 'I was so sad' #@param {type:"string"}
19
+ generator(text, num_return_sequences=1)
20
+
21
+ !pip install gradio
22
+
23
+ import gradio as gr
24
+ from transformers import pipeline
25
+
26
+ generator = pipeline('text-generation', model='huggingtweets/arthur_rimbaud')
27
+
28
+ def generate_text(input_text):
29
+ generated_text = generator(input_text, num_return_sequences=1)[0]['generated_text']
30
+ return generated_text
31
+
32
+ iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Rim-bot vuelve a casa")
33
+ iface.launch()
34
+