stop

app.py

@@ -8,7 +8,8 @@ def generate_text(prompt):
     # Load the model within the function so that it only runs on GPU when the function is called
     model = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", device=0)
 
-    return model(prompt, max_length=150, temperature=0.7, top_p=0.9)[0]["generated_text"]
+    return model(prompt, max_length=150, temperature=0.7, top_p=0.9, stop=["."])[0]["generated_text"]
+
 
 
 # Create the Gradio interface