Spaces:

deepapaikar
/

Katzbot-Phi2

App Files Files Community

deepapaikar commited on Feb 19

Commit

350e3ac

•

1 Parent(s): dda9ab5

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import gradio as gr
 #Setting device to cuda
 torch.set_default_device("cuda")
-device = 'cuda' if torch.cuda.is_availble() else 'cpu'
 model = "deepapaikar/katzbot-phi2"

 #Setting device to cuda
 torch.set_default_device("cuda")
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model = "deepapaikar/katzbot-phi2"