Update app.py

app.py

@@ -27,18 +27,18 @@ def get_gpt2_pipeline():
 # SD v1.4
 def get_stable_diffusion_v14_pipeline():
   model_id = "CompVis/stable-diffusion-v1-4"
-  pipeline = StableDiffusionPipeline.from_pretrained(mode_id)
+  pipe = StableDiffusionPipeline.from_pretrained(mode_id)
   # pipeline = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True, revision="fp16", torch_dtype=torch.float16)
-  pipeline = pipeline.to(DEVICE)
+  pipe = pipe.to(DEVICE)
   torch.backends.cudnn.benchmark = True
-  return pipeline
+  return pipe
 
 # SD v1.5
 def get_stable_diffusion_v15_pipeline():
   model_id = "runwayml/stable-diffusion-v1-5"
-  pipeline = DiffusionPipeline.from_pretrained(mode_id)
-  pipeline = pipeline.to(DEVICE)
-  return pipeline
+  pipe = DiffusionPipeline.from_pretrained(mode_id)
+  pipe = pipe.to(DEVICE)
+  return pipe
 
 def get_image(url):
   response = requests.get(url)
@@ -50,9 +50,9 @@ def get_image(url):
 def main():
   prompt = "Hello world, I'm vizard,"
   
-  pipeline = get_gpt2_pipeline()
+  pipe = get_gpt2_pipeline()
   def greet(prompt):
-    return pipeline(prompt, max_length=50, num_return_sequences=3)
+    return pipe(prompt, max_length=50, num_return_sequences=3)
   
   ui = gr.Interface(
     fn=greet,
@@ -61,7 +61,7 @@ def main():
   )
   ui.launch()
   
-  pipeline = get_stable_diffusion_v15_pipeline()
-  images = pipeline(prompt).images
+  pipe2 = get_stable_diffusion_v15_pipeline()
+  images = pipe2(prompt).images
 
 main