stefanola commited on
Commit
49249b8
·
1 Parent(s): f85d0df

Add application file

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ generator = pipeline('text-generation', model='stefanola/bloom560ftrmh')
5
+
6
+ def generate(text, full_text, new_tokens):
7
+ result = generator(text, num_return_sequences=1, return_full_text=full_text, max_new_tokens=new_tokens)
8
+ return result[0]["generated_text"]
9
+
10
+ examples = [
11
+ ["Höfuðborg Noregs er: Osló.\nHöfuðborg Englands er: London.\nHöfuðborg Íslands er:"],
12
+ ["Þessi mynd var frábær og ég mæli með henni fyrir alla | jákvætt\nMyndin var léleg og leikurinn slæmur | neikvætt\nLeikstjórn tókst vel til í myndinni og kom mjög á óvart | jákvætt\nLeikurinn var slappur og sagan leiðinleg | neikvætt\nÞessi mynd var nokkuð góð og tónlistin skemmtileg |"],
13
+ ["Frábær þjónusta, ég mæli með Landsbankanum! | jákvætt\nLandsbankinn er versti banki á Íslandi. Fékk mjög lélega þjónustu. | neikvætt\nÞað kom skemmtilega á óvart hvað starfsmenn voru liðtækir | jákvætt\nHæg þjónusta, þeir reyna bara að ræna af manni pening. | neikvætt\nÉg var frekar sáttur með þjónustuna í Landsbankanum |"]
14
+ ]
15
+
16
+ demo = gr.Interface(
17
+ fn=generate,
18
+ inputs=[
19
+ gr.inputs.Textbox(lines=5, label="Inntak"),
20
+ #gr.inputs.Number(default=100, label="Lengd"),
21
+ gr.inputs.Checkbox(default=False, label="Heildartexti"),
22
+ gr.inputs.Number(default=30, label="Fjöldi nýrra tóka")
23
+ ],
24
+ outputs=gr.outputs.Textbox(label="Úttak"),
25
+ examples=examples
26
+ )
27
+
28
+ demo.launch(share=True)