yiren98 commited on
Commit
536110b
·
1 Parent(s): 2a9d70b
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -116,7 +116,7 @@ def load_target_model(selected_model):
116
  lora_model.eval()
117
 
118
  logger.info("Models loaded successfully.")
119
- return "Models loaded successfully."
120
 
121
  except Exception as e:
122
  logger.error(f"Error loading models: {e}")
 
116
  lora_model.eval()
117
 
118
  logger.info("Models loaded successfully.")
119
+ return "Models loaded successfully. Using Recraft: {}".format(selected_model)
120
 
121
  except Exception as e:
122
  logger.error(f"Error loading models: {e}")