Update app.py

app.py

@@ -37,13 +37,14 @@ def txt2story(img_text):
     print("Story...")
     chat_completion = client.chat.completions.create(
         messages=messages,
+        "top_k": top_k,
+        "top_p": top_p,
+        "temperature": temperature,
         model="togethercomputer/llama-2-70b-chat")
     
     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...")
@@ -65,6 +66,11 @@ def main():
 # Allows users to upload an image file
     uploaded_file = st.file_uploader("# 📷 Upload an image...", type=["jpg", "jpeg", "png"])
 
+# Parameters for LLM model (in the sidebar)
+    top_k = st.sidebar.number_input("Top-K", min_value=1, max_value=100, value=5)
+    top_p = st.sidebar.number_input("Top-P", min_value=0.0, max_value=1.0, value=0.4)
+    temperature = st.sidebar.number_input("Temperature", min_value=0.1, max_value=2.0, value=1.0)
+
     if uploaded_file is not None:
 # Reads and saves uploaded image file
         bytes_data = uploaded_file.read()