Changed app.py

app.py

@@ -9,6 +9,7 @@ import pickle
 
 import torchvision.transforms as transforms
 from PIL import Image
+import os
 
 network_pkl_a = 'stylegan3-r-afhqv2-512x512.pkl'
 network_pkl_d = 'network-snapshot.pkl'
@@ -17,9 +18,11 @@ with open(network_pkl_d, 'rb') as f:
 
 with open(network_pkl_a, 'rb') as f:
     G_a = pickle.load(f)['G_ema'].cpu()  # torch.nn.Module
-    
-def gen_image(text):
-    if text=='show me the thing':
+
+cl_text = os.getenv('SHOW_TEXT')
+  
+def gen_image(text):    
+    if text==cl_text:
         return gen_image_helper(G_d)
     else:
         return gen_image_helper(G_a)