test

app.py

@@ -16,7 +16,7 @@ def create_prompt(stars, useful, funny, cool):
 
 def postprocess(review):
     dot = review.rfind('.')
-    return review[:dot]
+    return review[:dot+1]
 
 def generate_reviews(stars, useful, funny, cool):
     text = create_prompt(stars, useful, funny, cool)
@@ -31,7 +31,7 @@ def generate_reviews(stars, useful, funny, cool):
     )
     reviews = []
     for review in out:
-        reviews.append(postprocess(tokenizer.decode(review, skip_special_tokens=True)))
+        reviews.append(postprocess(tokenizer.decode(review)))#, skip_special_tokens=True)))
 
     return reviews[0], reviews[1], reviews[2]