pandora-s commited on
Commit
b26fd1f
·
verified ·
1 Parent(s): 305e0ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -50,7 +50,11 @@ for model in available_models:
50
 
51
  @spaces.GPU(duration=45)
52
  def run_inference(message, history, model_picked):
 
 
 
53
  local_dir = dirs[model_picked]
 
54
  print(message)
55
  print(history)
56
  # Loading only once GPU available
 
50
 
51
  @spaces.GPU(duration=45)
52
  def run_inference(message, history, model_picked):
53
+ if model_picked == None:
54
+ model_picked = default_bpw
55
+
56
  local_dir = dirs[model_picked]
57
+
58
  print(message)
59
  print(history)
60
  # Loading only once GPU available