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