codewithdark commited on
Commit
ca17086
·
verified ·
1 Parent(s): b7abf3f

Update utility/image_generator.py

Browse files
Files changed (1) hide show
  1. utility/image_generator.py +1 -8
utility/image_generator.py CHANGED
@@ -1,19 +1,12 @@
1
- from diffusers import DiffusionPipeline
2
- import torch
3
  import re
4
  from PIL import Image
5
  import io
6
  from dotenv import load_dotenv
7
  import os
 
8
 
9
  load_dotenv()
10
 
11
- # Ensure GPU is used if available
12
- device = "cuda" if torch.cuda.is_available() else "cpu"
13
-
14
-
15
- pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
16
- pipeline = pipeline.to(device)
17
 
18
  def generate_image_prompts(script):
19
  # Split the script into sentences
 
 
 
1
  import re
2
  from PIL import Image
3
  import io
4
  from dotenv import load_dotenv
5
  import os
6
+ import requests
7
 
8
  load_dotenv()
9
 
 
 
 
 
 
 
10
 
11
  def generate_image_prompts(script):
12
  # Split the script into sentences