refine generated text

app.py

@@ -194,7 +194,7 @@ def generate_captions(prompt, num_captions=3):
     for _ in range(num_captions):
         response = requests.post(API_URL, headers=headers, json=payload)
         output = response.json()
-        generated_text = output[0]['generated_text'].replace(prompt, '')
+        generated_text = output[0]['generated_text'].replace(prompt, '').split('.')[0] + '.'
         bloom_results.append(generated_text)
     return bloom_results