measmonysuon commited on
Commit
5533905
1 Parent(s): 8e3960a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -8
app.py CHANGED
@@ -113,14 +113,6 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
113
  # Always request to switch back hardware
114
  api.request_space_hardware(repo_id='measmonysuon/DALLE-4K', hardware='zero-a10g')
115
  logger.info("Hardware reset to Zero-A10G")
116
-
117
- finally:
118
- # Log GPU info before resetting hardware
119
- logger.info(f"GPU info before reset: {gpu_info}")
120
-
121
- # Always request to switch back hardware
122
- api.request_space_hardware(repo_id='measmonysuon/DALLE-4K', hardware='zero-a10g')
123
- logger.info("Hardware reset to Zero-A10G")
124
 
125
  def handle_generate_image(prompt, resolution_key, user_chat_id):
126
  points = get_user_points(user_chat_id)
 
113
  # Always request to switch back hardware
114
  api.request_space_hardware(repo_id='measmonysuon/DALLE-4K', hardware='zero-a10g')
115
  logger.info("Hardware reset to Zero-A10G")
 
 
 
 
 
 
 
 
116
 
117
  def handle_generate_image(prompt, resolution_key, user_chat_id):
118
  points = get_user_points(user_chat_id)