update space

app.py

@@ -22,7 +22,7 @@ else:
 print("Setting up MongoDB connection...")
 MONGO_URI = os.environ.get("MONGO_URI")
 client = MongoClient(MONGO_URI)
-db = client.poultry_farm  # Database
+db = client.poultry_management  # Database
 print("MongoDB connection established.")
 
 # Check GPU availability for TensorFlow
@@ -171,11 +171,11 @@ def llama2_response(user_input):
     try:
         print("Generating response using Llama 2...")
         # Tokenize user input for the Llama 2 model
-        inputs = tokenizer(user_input, return_tensors="pt", truncation=True, max_length=150, padding=True)
+        inputs = tokenizer(user_input, return_tensors="pt", truncation=True, max_length=500, padding=True)
         # Generate a response using the Llama 2 model
         outputs = model.generate(
             inputs["input_ids"],
-            max_length=150,
+            max_length=500,
             do_sample=True,
             temperature=0.7,
             pad_token_id=tokenizer.pad_token_id,  # Use the newly set padding token