Update app.py

app.py

@@ -50,16 +50,19 @@ def invoke(prompt):
         #                                      temperature = config["temperature"],
         #                                      top_k = config["top_k"],
         #                                      top_p = config["top_p"],
-        #                                     ).text
+        #                                     )
+        #if (completion.text != None):
+        #    completion = completion.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
+                                                               })
+        if (completion.text != None):
+            completion = completion.text
     except Exception as e:
         completion = e
-        print(completion)
         raise gr.Error(e)
     finally:
         wandb_log(prompt, completion)