Update app.py

app.py

@@ -1,7 +1,10 @@
 import streamlit as st
 import os
+import requests
+import io
+from PIL import Image
 
-hf_token = os.environ("API_TOKEN")
+hf_token = os.environ.get("API_TOKEN")
 headers = {"Authorization": f"Bearer {hf_token}"}
 API_URL_FACE = "https://api-inference.huggingface.co/models/prompthero/linkedin-diffusion"
 API_URL_PIX = "https://api-inference.huggingface.co/models/nerijs/pixel-art-xl"
@@ -30,7 +33,7 @@ elif model == '3D Render':
 elif model == 'Realistic':
     API_URL = API_URL_REAL
 elif model == 'Inkpunk':
-    API_URL = API_URL_INKPUNk
+    API_URL = API_URL_INKPUNK
 elif model == 'Dremscape':
     API_URL = API_URL_DREAM
 elif model == 'Magazine-cover':
@@ -40,12 +43,12 @@ elif model == 'Faces':
 
 
 def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.content
+  response = requests.post(API_URL, headers=headers, json=payload)
+  return response.content
     
 def generate_image(input_text):
     image_bytes = query({
-        "inputs": prompt,
+        "inputs": input_text,
     })
     image = Image.open(io.BytesIO(image_bytes))
     return image