measmonysuon commited on
Commit
e84b948
1 Parent(s): a855a97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -3,8 +3,7 @@ from gradio_client import Client
3
  import os
4
  import logging
5
  import requests
6
- from huggingface_hub import login
7
- from huggingface_hub import whoami
8
 
9
  webhook_server = os.getenv('webhook_server')
10
  API_TOKEN = os.getenv('ZeroGPU')
@@ -19,6 +18,10 @@ login(token=API_TOKEN, add_to_git_credential=True)
19
  # Initialize the client for image generation "prithivMLmods/IMAGINEO-4K" "mukaist/DALLE-4K"
20
  client_image = Client("measmonysuon/DALLE-4K")
21
 
 
 
 
 
22
  # Define resolutions and default style
23
  resolutions = {
24
  "896x1152": (896, 1152),
 
3
  import os
4
  import logging
5
  import requests
6
+ from huggingface_hub import login, whoami
 
7
 
8
  webhook_server = os.getenv('webhook_server')
9
  API_TOKEN = os.getenv('ZeroGPU')
 
18
  # Initialize the client for image generation "prithivMLmods/IMAGINEO-4K" "mukaist/DALLE-4K"
19
  client_image = Client("measmonysuon/DALLE-4K")
20
 
21
+ # Get user information to confirm authentication
22
+ user_info = whoami(token=API_TOKEN)
23
+ print("User Info:", user_info)
24
+
25
  # Define resolutions and default style
26
  resolutions = {
27
  "896x1152": (896, 1152),