kz209 commited on
Commit
5275a8d
β€’
1 Parent(s): 57ace24
Files changed (1) hide show
  1. pages/summarization_playground.py +1 -1
pages/summarization_playground.py CHANGED
@@ -10,7 +10,7 @@ import torch
10
 
11
  load_dotenv()
12
 
13
- __model_on_gpu__ = Model.__model_list__[0]
14
  model = {model_name: None for model_name in Model.__model_list__}
15
 
16
  random_label = 'πŸ”€ Random dialogue from dataset'
 
10
 
11
  load_dotenv()
12
 
13
+ __model_on_gpu__ = ''
14
  model = {model_name: None for model_name in Model.__model_list__}
15
 
16
  random_label = 'πŸ”€ Random dialogue from dataset'