Update app.py

app.py

@@ -21,7 +21,7 @@ def img2txt(url):
     return text
 
 # Text-to-story
-def txt2story(img_text):
+def txt2story(img_text, top_k, top_p, temperature):
     print("Initializing client...")
     client = openai.OpenAI(
         api_key=os.environ["TOGETHER_API_KEY"], 
@@ -85,7 +85,7 @@ def main():
         # Initiates AI processing and story generation
         with st.spinner("## 🤖 AI is at Work! "):
             scenario = img2txt("uploaded_image.jpg")  # Extracts text from the image
-            story = txt2story(scenario)  # Generates a story based on the image text
+            story = txt2story(scenario, top_k, top_p, temperature)  # Generates a story based on the image text, LLM params
             txt2speech(story)  # Converts the story to audio
 
             st.markdown("---")