bug fix

Signed-off-by: peter szemraj <[email protected]>

app.py

@@ -82,15 +82,13 @@ def load_emailgen_model(model_tag:str):
     Returns:
         transformers.pipelines.TextGenerationPipeline: the text generation pipeline
     """
-
+    global generator
     generator = pipeline(
         "text-generation",
         model_tag,
         device=0 if use_gpu else -1,
     )
 
-    return generator
-
 def get_parser():
     """
     get_parser - a helper function for the argparse module
@@ -248,7 +246,7 @@ if __name__ == "__main__":
         load_model_button.click(
             fn=load_emailgen_model,
             inputs=[model_name],
-            outputs=[generator],
+            outputs=[],
         )
     demo.launch(
         enable_queue=True,