Update app.py

app.py

@@ -21,12 +21,12 @@ def img2txt(url):
     return text
 
 # Text-to-story
-def txt2story(img_text, llm_model, top_k, top_p, temperature):
+def txt2story(img_text, top_k, top_p, temperature):
 
     headers = {"Authorization": f"Bearer {os.environ['TOGETHER_API_KEY']}"}
 
     data = {
-        "model": llm_model,
+        "model": togethercomputer/llama-2-70b-chat,
         "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": temperature},