TakiTakiTa commited on
Commit
86a6ee8
·
verified ·
1 Parent(s): ec40b9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -27,6 +27,8 @@ def load_model(model_name: str):
27
  @spaces.GPU
28
  def generate(prompt, history):
29
  global loaded_models, current_model_name
 
 
30
  if current_model_name == "" or current_model_name not in loaded_models:
31
  return "Please load a model first by entering a model name and clicking the Load Model button."
32
 
 
27
  @spaces.GPU
28
  def generate(prompt, history):
29
  global loaded_models, current_model_name
30
+ print("loaded models: ", loaded_models)
31
+ print("current model: ", current_model_name)
32
  if current_model_name == "" or current_model_name not in loaded_models:
33
  return "Please load a model first by entering a model name and clicking the Load Model button."
34