maheer commited on
Commit
01b63b3
·
verified ·
1 Parent(s): 0e26dc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,12 +7,13 @@ st.set_page_config(
7
  page_icon=":robot:",
8
  layout='wide'
9
  )
10
- device = "cuda:0"
11
  st.title("Creek")
12
 
13
  @st.cache_resource
14
  def get_model():
15
  import os
 
16
  os.system('pip install transformers')
17
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
18
  class FastapiTaskStramer(TextIteratorStreamer):
 
7
  page_icon=":robot:",
8
  layout='wide'
9
  )
10
+ device = "cpu"
11
  st.title("Creek")
12
 
13
  @st.cache_resource
14
  def get_model():
15
  import os
16
+ os.system('pip install torch')
17
  os.system('pip install transformers')
18
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
19
  class FastapiTaskStramer(TextIteratorStreamer):