futranbg commited on
Commit
5bdd91c
·
1 Parent(s): 360f04a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -61,7 +61,8 @@ def generate(
61
 
62
  cli = client
63
  status = cli.get_model_status()
64
- if not status.loaded:
 
65
  cli = client2
66
 
67
  try:
 
61
 
62
  cli = client
63
  status = cli.get_model_status()
64
+ if not status.loaded and status.state == 'Loadable':
65
+ cli.text_generation('Hello!', **generate_kwargs, return_full_text=False)
66
  cli = client2
67
 
68
  try: