Update app.py

app.py

@@ -4,14 +4,12 @@ from PIL import Image
 from io import BytesIO
 import os
 
-token = os.environ['AUTH_TOKEN']
-runpod_id = os.environ['RUNPOD_ID']
 
-url = 'https://{runpod_id}-8000.proxy.runpod.net/generate-image/'
+url = 'https://{os.environ['RUNPOD_ID']}-8000.proxy.runpod.net/generate-image/'
 
-def generate_image_from_text(prompt, class_guidance, token):
+def generate_image_from_text(prompt, class_guidance):
     headers = {
-        'Authorization': f'Bearer {token}'
+        'Authorization': f'Bearer {os.environ['AUTH_TOKEN']}'
     }
     
     data = {