Update app.py

app.py

@@ -21,7 +21,11 @@ def img2txt(url):
     return text
 
 # Text-to-story
-data = {
+def txt2story(img_text, llm_model, top_k, top_p, temperature):
+
+    headers = {"Authorization": f"Bearer {os.environ['TOGETHER_API_KEY']}"}
+
+    data = {
         "model": llm_model,
         "messages": [
             {"role": "system", "content": '''As an experienced short story writer, write story title and then create a meaningful story influenced by provided words.