Update app.py

app.py

@@ -8,7 +8,7 @@ import json
 @st.cache(allow_output_mutation=True)
 def load_pipeline(model_ckpt, device):
     pipe = pipeline('text-generation', model=model_ckpt, device=device)
-    return model
+    return pipe
 
 @st.cache()
 def load_examples():