update app

app.py

@@ -14,28 +14,32 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_toke
 def create_prompt(stars, useful, funny, cool):
     return f"Generate review: stars: {stars}, useful: {useful}, funny: {funny}, cool: {cool}"
 
+def postprocess(review):
+    dot = review.rfind('.')
+    return review[:dot]
+
 def generate_reviews(stars, useful, funny, cool):
     text = create_prompt(stars, useful, funny, cool)
     inputs = tokenizer(text, return_tensors='pt')
     out = model.generate(
         input_ids=inputs.input_ids,
         attention_mask=inputs.attention_mask,
-        num_beams=5,
-        num_return_sequences=3
+        num_beams=3,
+        num_return_sequences=3,
+        temperature=1.2
     )
     reviews = []
     for review in out:
-        reviews.append(tokenizer.decode(review, skip_special_tokens=True))
+        reviews.append(postprocess(tokenizer.decode(review, skip_special_tokens=True)))
 
     return reviews[0], reviews[1], reviews[2]
 
 css = """
     #ctr {text-align: center;}
-    #btn {color: white; background: linear-gradient(90deg, #00d2ff 0%, #3a47d5 100%);}
+    #btn {color: white; background: linear-gradient( 90deg,  rgba(255,166,0,1) 14.7%, rgba(255,99,97,1) 73% );}
 """
 
 
-
 md_text = """## Generating Yelp reviews with BART-base ⭐⭐⭐"""
 demo = gr.Blocks(css=css)
 with demo: