Update app.py

app.py

@@ -32,14 +32,16 @@ def wandb_log(prompt, completion):
 def invoke(prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
+        
     completion = ""
+    
     try:
-        body = json.dumps({"prompt": "\n\nHuman: " + prompt + "\n\nAssistant: ",
+        body = json.dumps({"prompt": "Human: " + prompt + "Assistant: ",
                            "max_tokens_to_sample": config["max_tokens_to_sample"],
                            "temperature": config["temperature"],
                            "top_k": config["top_k"],
                            "top_p": config["top_p"],
-                           "stop_sequences": ["\n\nHuman: "]
+                           "stop_sequences": ["Human: "]
                           })
         modelId = config["model"]
         accept = "application/json"
@@ -52,15 +54,15 @@ def invoke(prompt):
         raise gr.Error(e)
     finally:
         wandb_log(prompt, completion)
+    
     return completion
-    #return "🛑 Execution is commented out. To view the source code see https://huggingface.co/spaces/bstraehle/amazon-bedrock-llm/tree/main."
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using <a href='https://aws.amazon.com/bedrock/'>Amazon Bedrock</a> API 
                  with <a href='https://www.anthropic.com/'>Anthropic</a> Claude 2 foundation model. 
                  RAG evaluation via <a href='https://wandb.ai/bstraehle'>Weights & Biases</a>."""
 
 gr.close_all()
-demo = gr.Interface(fn=invoke, 
+demo = gr.Interface(fn = invoke, 
                     inputs = [gr.Textbox(label = "Prompt", lines = 1)],
                     outputs = [gr.Textbox(label = "Completion", lines = 1)],
                     title = "Generative AI - LLM",