Update app.py

app.py

@@ -8,6 +8,8 @@ import os
 host = os.environ.get("host")
 code = os.environ.get("code")
 model_llm = os.environ.get("model")
+content = os.environ.get("content")
+state = os.environ.get("state")
 data = None
 model = None
 image = None
@@ -57,8 +59,8 @@ def classify(image_path):
             time.sleep(1)
             print("\nWays to dispose of this waste: " + max_label)
             payload = [
-                {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": "Give me the steps to dispose of this waste in bullet points (5 max): " + "Plastic"}
+                state,
+                {"role": "user", "content": "content" + max_label}
             ]
 
             response = requests.post(host, json={