Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 
 generator = pipeline('text-generation', model='stefanola/bloom560ftrmh')
 
-def generate(text):#, full_text, new_tokens):
-    result = generator(text, num_return_sequences=1)#, return_full_text=full_text, max_new_tokens=new_tokens)
+def generate(text, new_tokens):
+    result = generator(text, num_return_sequences=1, return_full_text=False, max_new_tokens=new_tokens)
     return result[0]["generated_text"]
 
 examples = [
@@ -16,10 +16,10 @@ examples = [
 demo = gr.Interface(
     fn=generate,
     inputs=[
-        gr.inputs.Textbox(lines=5, label="Inntak")
+        gr.inputs.Textbox(lines=5, label="Inntak"),
         #gr.inputs.Number(default=100, label="Lengd"),
         #gr.inputs.Checkbox(default=False, label="Heildartexti"),
-        #gr.inputs.Number(default=30, label="Fjöldi nýrra tóka")
+        gr.inputs.Number(default=30, label="Fjöldi úttakstóka")
     ],
     outputs=gr.outputs.Textbox(label="Úttak"),
     examples=examples