Spaces:
Sleeping
Sleeping
removed pipeline
Browse files
app.py
CHANGED
@@ -40,10 +40,9 @@ Keep your responses concise. If the request is unethical or out of scope, polite
|
|
40 |
@st.cache_resource
|
41 |
def load_model_controller():
|
42 |
# Controller: microsoft/Phi-3-mini-4k-instruct
|
43 |
-
pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
|
44 |
tokenizerC = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
|
45 |
modelC = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
|
46 |
-
return tokenizerC, modelC
|
47 |
|
48 |
@st.cache_resource
|
49 |
def load_model_engineer():
|
|
|
40 |
@st.cache_resource
|
41 |
def load_model_controller():
|
42 |
# Controller: microsoft/Phi-3-mini-4k-instruct
|
|
|
43 |
tokenizerC = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
|
44 |
modelC = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
|
45 |
+
return tokenizerC, modelC
|
46 |
|
47 |
@st.cache_resource
|
48 |
def load_model_engineer():
|