Update app.py

app.py

@@ -41,9 +41,9 @@ generation_model = GenerativeModel(config.model)
 
 def invoke(prompt):
     #completion = generation_model.predict(prompt = prompt).text
-    #completion = generation_model.generate_content(prompt, stream = False).text
-    print(generation_model.generate_content(prompt))
-    print(generation_model.generate_content(prompt).text)
+    completion = generation_model.generate_content(prompt).text
+    #print(generation_model.generate_content(prompt))
+    #print(generation_model.generate_content(prompt).text)
     wandb.log({"prompt": prompt, "completion": completion})
     return completion
     return "Execution is commented out, but you can view the code at https://huggingface.co/spaces/bstraehle/google-vertex-ai-llm/tree/main."