Update app.py

app.py

@@ -51,14 +51,14 @@ class Story(Document):
     story_id = SequenceField(primary_key=True)
     
 def make_prompt(entry):
-    return f"### Human: YOUR INSTRUCTION HERE,ALWAYS USE A STORY,INCLUDE ASSESMENTS,TECHNICAL SUMMARY: {entry} ### Assistant:"
+    return f"### Human: YOUR INSTRUCTION HERE,ALWAYS USE A STORY,INCLUDE ASSESMENTS THROUGHOUT AND A TECHNICAL SUMMARY: {entry} ### Assistant:"
     
 @spaces.GPU
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],
     max_new_tokens: int = 1024,
-    temperature: float = 0.1,  # Lower -> less random
+    temperature: float = 0.3,  # Lower -> less random
     top_p: float = 0.1,  # Lower -> less random, considering only the top 10% of tokens at each step
     top_k: int = 1,  # Least random, only the most likely next token is considered
     repetition_penalty: float = 1.0,  # No repetition penalty
@@ -96,6 +96,7 @@ def generate(
     for text in streamer:
         outputs.append(text)
         yield "".join(outputs)
+    final_story = "".join(outputs)
     try:
         saved_story = Story(message=message, content=final_story).save() 
         yield f"Story saved with ID: {saved_story.story_id}"
@@ -119,5 +120,6 @@ with gr.Blocks(css="style.css") as demo:
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch()
+    demo.launch(share=True)