Update app.py

app.py

@@ -163,7 +163,7 @@ def create_vector_db(final_items):
         Here is the antimony segment to summarize: {item}
         """
         inputs = tokenizer.encode(prompt, return_tensors="pt").to(device)
-        response = model.generate(inputs)
+        response = model.generate(inputs, max_length = 100000000000)
         documents.append(tokenizer.decode(response[0]))
     
     if final_items:
@@ -177,7 +177,7 @@ def create_vector_db(final_items):
 def generate_response(db, query_text, previous_context):
     query_results = db.query(
         query_texts=query_text,
-        n_results=5,
+        n_results=15,
     )
     
     if not query_results.get('documents'):