Update app.py

app.py

@@ -30,21 +30,23 @@ def txt2story(img_text):
     
     messages = [
         {"role": "system", "content": '''As an experienced short story writer, write story title and then create a meaningful story influenced by provided words. 
-        Ensure stories conclude positively within 100 words. Remember the story must end within 100 words''', "temperature": 1.8},
-        {"role": "user", "content": f"Here is input set of words: {img_text}", "temperature": 1.5},
+        Ensure stories conclude positively within 100 words. Remember the story must end within 100 words''', "temperature": temperature},
+        {"role": "user", "content": f"Here is input set of words: {img_text}", "temperature": temperature},
     ]
     
     print("Story...")
     chat_completion = client.chat.completions.create(
+        model="togethercomputer/llama-2-70b-chat",
         messages=messages,
-        "top_k": top_k,
-        "top_p": top_p,
-        "temperature": temperature,
-        model="togethercomputer/llama-2-70b-chat")
+        top_k=top_k,
+        top_p=top_p,
+        temperature=temperature
+    )
     
     print(chat_completion.choices[0].message.content)
     return chat_completion.choices[0].message.content
 
+
 # Text-to-speech
 def txt2speech(text):
     print("Initializing text-to-speech conversion...")