Update app.py

app.py

@@ -30,7 +30,7 @@ def index():
                 os.environ.get('HF_MODEL_ID'),
                 token=random.choice(list_of_token)
             )
-            prompt = f"""{os.environ.get('HF_PROMPT')}"""
+            prompt = f"""{os.environ.get('HF_PROMPT')}\n{code}\n{os.environ.get('HF_PROMPT2')}"""
             res = client.text_generation(prompt, max_new_tokens=1024, stop=["```", "</Code>"])
             res = res[:res.rfind('\n')]
             return jsonify({'result': res})