Update app.py

app.py

@@ -12,8 +12,9 @@ from themes import IndonesiaTheme  # Import custom IndonesiaTheme
 from lorify import Lorify
 from css import css2
 MAX_SEED = np.iinfo(np.int32).max
-HF_TOKEN = "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"
-HF_TOKEN_UPSCALER = "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"
+
+HF_TOKEN = os.getenv('HF_TOKEN')
+
 
 HF_TOKEN = os.getenv('HF_TOKEN_UPSCALER')
 HF_TOKEN_UPSCALER = os.getenv('HF_TOKEN_UPSCALER')
@@ -50,7 +51,7 @@ async def generate_image(prompt, model, lora_word, width, height, scales, steps,
 def get_upscale_finegrain(prompt, img_path, upscale_factor):
     try:
         print(f"[-] Memulai proses upscaling dengan faktor {upscale_factor} untuk gambar {img_path}")
-        client = Client("finegrain/finegrain-image-enhancer", hf_token=HF_TOKEN_UPSCALER)
+        client = Client("finegrain/finegrain-image-enhancer", hf_token=HF_TOKEN)
         result = client.predict(
             input_image=handle_file(img_path), 
             prompt=prompt,