Rakesh443 commited on
Commit
1099382
1 Parent(s): 82ca5d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
51
  API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
52
 
53
 
54
- api_key = os.getenv("Imagrating")
55
- headers = {"Authorization": "f'Bearer {api_key}"}
56
 
57
 
58
  if image_style == "None style":
 
51
  API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
52
 
53
 
54
+ API_KEY = os.environ.get("API_KEY")
55
+ headers = {"Authorization": f"Bearer {API_KEY}"}
56
 
57
 
58
  if image_style == "None style":