danidanidani commited on
Commit
6057e5a
1 Parent(s): e1d39a6

Update src/backend/chatbot.py

Browse files
Files changed (1) hide show
  1. src/backend/chatbot.py +0 -5
src/backend/chatbot.py CHANGED
@@ -33,11 +33,6 @@ print("BP 4 ")
33
  def init_llm(model, demo_lite):
34
  # st.write("BP 4.1: model: ", model)
35
  if demo_lite == False:
36
- api = HfApi(token=HF_TOKEN)
37
- runtime = api.get_space_runtime(repo_id=TRAINING_SPACE_ID)
38
- # Check if Space is loaded with a GPU.
39
- if runtime.hardware == SpaceHardware.T4_SMALL:
40
- print("using nvidia T4 small")
41
 
42
  print("BP 5 : running full demo")
43
  if model == "Llama2-7b_CPP":
 
33
  def init_llm(model, demo_lite):
34
  # st.write("BP 4.1: model: ", model)
35
  if demo_lite == False:
 
 
 
 
 
36
 
37
  print("BP 5 : running full demo")
38
  if model == "Llama2-7b_CPP":