apapiu commited on
Commit
7013fe5
1 Parent(s): 791b323

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,12 +4,14 @@ from PIL import Image
4
  from io import BytesIO
5
  import os
6
 
 
 
7
 
8
- url = f'https://{os.environ['RUNPOD_ID']}-8000.proxy.runpod.net/generate-image/'
9
 
10
  def generate_image_from_text(prompt, class_guidance):
11
  headers = {
12
- 'Authorization': f'Bearer {os.environ['AUTH_TOKEN']}'
13
  }
14
 
15
  data = {
 
4
  from io import BytesIO
5
  import os
6
 
7
+ runpod_id = os.environ['RUNPOD_ID']
8
+ token_id = os.environ['AUTH_TOKEN']
9
 
10
+ url = f'https://{runpod_id}-8000.proxy.runpod.net/generate-image/'
11
 
12
  def generate_image_from_text(prompt, class_guidance):
13
  headers = {
14
+ 'Authorization': f'Bearer {token_id}'
15
  }
16
 
17
  data = {