Update app.py

app.py

@@ -37,6 +37,8 @@ def generate_input(input_type, image=None, text=None, response_amount=3):
     # If no input, fallback
     if not combined_input:
         combined_input = "No input provided."
+    if response_amount is None:
+        response_amount=3
 
     return vector_search(combined_input,response_amount)
 
@@ -48,6 +50,8 @@ metadata = pd.read_csv("netflix_metadata.csv") #created using sentence_transform
 def vector_search(query,top_n=3):
     query_embedding = sentence_model.encode(query)
     similarities = cosine_similarity([query_embedding], embeddings)[0]
+    if top_n is None:
+        top_n=3
     top_indices = similarities.argsort()[-top_n:][::-1]
     results = metadata.iloc[top_indices]
     result_text=""
@@ -90,6 +94,8 @@ with gr.Blocks() as demo:
     
     submit_button = gr.Button("Submit")
     output = gr.Textbox(label="Recommendations")
+    if selected_response_amount is None:
+        selected_response_amount=3
 
     submit_button.click(fn=generate_input, inputs=[input_type,image_input, text_input,selected_response_amount], outputs=output)
 demo.launch()