Spaces:

Kaushik066
/

indian_sign_language_translation

Running

Kaushik066 commited on Mar 18

Commit

6018306

1 Parent(s): 7d02469

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 MODEL_TRANSFORMER = 'google/vivit-b-16x2'
 # Set Paths
 #model_path = 'vivit_pytorch_loss051.pt'
-model_path_2_pytorch = 'vivit_pytorch_GPU_6_acc087.pt'
 #model_path_2_transformer = ''
 data_path = 'signs'

 MODEL_TRANSFORMER = 'google/vivit-b-16x2'
 # Set Paths
 #model_path = 'vivit_pytorch_loss051.pt'
+model_path_2_pytorch = 'models/vivit_ISL_pt_6_76classes_loss051.pt'
 #model_path_2_transformer = ''
 data_path = 'signs'