Spaces:

manan05
/

friends

Sleeping

manan05 commited on Oct 25, 2023

Commit

2939849

•

1 Parent(s): a6b1f94

works?

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,8 @@ generator = pipeline(
     model=model, tokenizer=tokenizer,
     task='text-generation',
     max_new_tokens=50,
-    repetition_penalty=1.1
 )
 with gr.Blocks() as demo:

     model=model, tokenizer=tokenizer,
     task='text-generation',
     max_new_tokens=50,
+    repetition_penalty=1.1,
+    return_full_text=False
 )
 with gr.Blocks() as demo: