Add util method to get image

app.py

@@ -4,6 +4,7 @@ import requests
 import gradio as gr
 from datasets import load_dataset
 from PIL import Image
+from io import BytesIO
 import torch
 from torch import autocast
 
@@ -35,7 +36,13 @@ def get_stable_diffusion_v15_pipeline():
   pipeline = DiffusionPipeline.from_pretrained(mode_id)
   pipeline = pipeline.to(DEVICE)
   return pipeline
-  
+
+def get_image(url):
+  response = requests.get(url)
+  image = Image.open(BytesIO(response.content)).convert("RGB")
+  resized_image = image.resize((768, 512))
+  return resized_image
+
 # main
 def main():
   prompt = "Hello world, I'm vizard,"