Update pages/type_text.py
Browse files- 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": []}
|