Update app.py

app.py

@@ -87,7 +87,7 @@ def generate_image(image, product_name, target_name):
   num_samples = 4
 
   prompt = target_name
-  generator = torch.Generator(device="cuda").manual_seed(22) # change the seed to get different results
+  generator = torch.Generator(device=device).manual_seed(22) # change the seed to get different results
 
   im = IPpipe(
       prompt=prompt,
@@ -123,7 +123,7 @@ for code in codes_as_string:
 import gradio as gr
 import gc 
 gc.collect()
-%env PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:256
+
 image_label = 'Please upload the image (optional)'
 extract_label = 'Specify what need to be extracted from the above image'
 prompt_label = 'Specify the description of image to be generated'