Ashrafb commited on
Commit
0bbd29b
1 Parent(s): b65b732

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -16,7 +16,7 @@ import base64
16
 
17
  app = FastAPI()
18
 
19
- API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
20
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
21
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
22
  def generate_image(prompt):
@@ -37,7 +37,7 @@ def generate_image(prompt):
37
 
38
 
39
  text_gen = gr.Interface.load("models/Gustavosta/MagicPrompt-Stable-Diffusion")
40
- proc1 = gr.Interface.load("models/playgroundai/playground-v2-1024px-aesthetic")
41
 
42
  queue = Queue()
43
  queue_threshold = 100
 
16
 
17
  app = FastAPI()
18
 
19
+ API_URL = "https://api-inference.huggingface.co/models/ehristoforu/dalle-3-xl"
20
  API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
21
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
22
  def generate_image(prompt):
 
37
 
38
 
39
  text_gen = gr.Interface.load("models/Gustavosta/MagicPrompt-Stable-Diffusion")
40
+ proc1 = gr.Interface.load("models/ehristoforu/dalle-3-xl")
41
 
42
  queue = Queue()
43
  queue_threshold = 100