GCBQDevTeam commited on
Commit
c331c0b
·
verified ·
1 Parent(s): fa4dfcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import os
6
 
7
  def greet(null):
8
  string = ""
9
- device = torch.device('cuda')
10
- model_list = ['tiny.en', 'base.en']#, 'medium.en', 'large-v2']
11
  fp16_bool = [True, False]
12
  path = 'benchmark/'
13
  file_list = os.listdir(path)
 
6
 
7
  def greet(null):
8
  string = ""
9
+ device = torch.device('cpu')
10
+ model_list = ['tiny.en', 'base.en', "small.en" 'medium', 'large-v2']
11
  fp16_bool = [True, False]
12
  path = 'benchmark/'
13
  file_list = os.listdir(path)