prithivMLmods commited on
Commit
9278cbb
·
verified ·
1 Parent(s): f191662

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -18
app.py CHANGED
@@ -1,4 +1,3 @@
1
- # ...
2
  import os
3
  import random
4
  import uuid
@@ -26,26 +25,11 @@ examples = [
26
  "Cold coffee in a cup bokeh --ar 85:128 --v 6.0 --style raw5, 4K",
27
  ]
28
 
29
-
30
- #examples = [
31
- # ["file/1.png", "3d image, cute girl, in the style of Pixar --ar 1:2 --stylize 750, 4K resolution highlights, Sharp focus, octane render, ray tracing, Ultra-High-Definition, 8k, UHD, HDR, (Masterpiece:1.5), (best quality:1.5)"],
32
- # ["file/2.png", "Cold coffee in a cup bokeh --ar 85:128 --v 6.0 --style raw5, 4K"],
33
- #["file/3.png", "Vector illustration of a horse, vector graphic design with flat colors on a brown background in the style of vector art, using simple shapes and graphics with simple details, professionally designed as a tshirt logo ready for print on a white background. --ar 89:82 --v 6.0 --style raw"],
34
- #["file/4.png", "Man in brown leather jacket posing for the camera, in the style of sleek and stylized, clockpunk, subtle shades, exacting precision, ferrania p30 --ar 67:101 --v 5"],
35
- #["file/5.png", "Commercial photography, giant burger, white lighting, studio light, 8k octane rendering, high resolution photography, insanely detailed, fine details, on a white isolated plain, 8k, commercial photography, stock photo, professional color grading, --v 4 --ar 9:16"]
36
- #]
37
-
38
-
39
- #Set an os.Getenv variable
40
- #set VAR_NAME=”VALUE”
41
- #Fetch an environment variable
42
- #echo %VAR_NAME%
43
-
44
- MODEL_ID = os.getenv("MODEL_VAL_PATH") #Use SDXL Model as "MODEL_REPO" --------->>> ”VALUE”.
45
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
46
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
47
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
48
- BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1")) # Allow generating multiple images at once
49
 
50
  #Load model outside of function
51
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 
1
  import os
2
  import random
3
  import uuid
 
25
  "Cold coffee in a cup bokeh --ar 85:128 --v 6.0 --style raw5, 4K",
26
  ]
27
 
28
+ MODEL_ID = os.getenv("MODEL_VAL_PATH", "SG161222/RealVisXL_V4.0_Lightning")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
30
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
31
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
32
+ BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
33
 
34
  #Load model outside of function
35
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")