Update app.py

app.py

@@ -48,15 +48,15 @@ def invoke(prompt):
     #                                      top_k = config["top_k"],
     #                                      top_p = config["top_p"],
     #                                     ).text
-    #completion = generation_model.generate_content(prompt, generation_config = {
-    #                                                           "max_output_tokens": config["max_output_tokens"],
-    #                                                           "temperature": config["temperature"],
-    #                                                           "top_k": config["top_k"],
-    #                                                           "top_p": config["top_p"],
-    #                                                       }).text
-    #wandb_log(prompt, completion)
-    #return completion
-    return "🛑 Execution is commented out. To view the source code see https://huggingface.co/spaces/bstraehle/google-vertex-ai-llm/tree/main."
+    completion = generation_model.generate_content(prompt, generation_config = {
+                                                               "max_output_tokens": config["max_output_tokens"],
+                                                               "temperature": config["temperature"],
+                                                               "top_k": config["top_k"],
+                                                               "top_p": config["top_p"],
+                                                           }).text
+    wandb_log(prompt, completion)
+    return completion
+    #return "🛑 Execution is commented out. To view the source code see https://huggingface.co/spaces/bstraehle/google-vertex-ai-llm/tree/main."
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using <a href='https://cloud.google.com/vertex-ai?hl=en/'>Google Vertex AI</a> API 
                  with gemini-pro foundation model. RAG evaluation via <a href='https://wandb.ai/bstraehle'>Weights & Biases</a>."""