georad commited on
Commit
73fcf1f
·
verified ·
1 Parent(s): a153606

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +2 -1
pages/type_text.py CHANGED
@@ -77,7 +77,8 @@ HF_model_results_displayed = HF_model_results_sorted[0:numMAPPINGS_input]
77
  @st.cache_resource
78
  def load_model_pipe():
79
  pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B-Instruct", device_map=device,) # device_map="auto", torch_dtype=torch.bfloat16
80
- return pipe
 
81
  load_model_pipe()
82
 
83
  dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}
 
77
  @st.cache_resource
78
  def load_model_pipe():
79
  pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B-Instruct", device_map=device,) # device_map="auto", torch_dtype=torch.bfloat16
80
+ return pipe
81
+ pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B-Instruct", device_map=device,) # device_map="auto", torch_dtype=torch.bfloat16
82
  load_model_pipe()
83
 
84
  dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}