Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|