Hugging Face's logo

Spaces:
Vishnu-add
/
Search-your-doc
Sleeping

App Files Community
Vishnu-add commited on
Commit
b33c6ec
•
1 Parent(s): bb6a0e4

Changed the device

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ from langchain.chains import RetrievalQA
11
 
12
  @st.cache_resource
13
  def get_model():
14
- # device = torch.device('cpu')
15
- device = torch.device('cuda:0')
16
 
17
  checkpoint = "LaMini-T5-738M"
18
  checkpoint = "MBZUAI/LaMini-T5-738M"
 
11
 
12
  @st.cache_resource
13
  def get_model():
14
+ device = torch.device('cpu')
15
+ # device = torch.device('cuda:0')
16
 
17
  checkpoint = "LaMini-T5-738M"
18
  checkpoint = "MBZUAI/LaMini-T5-738M"