6Genix commited on
Commit
01bc7e8
·
1 Parent(s): abd0aa0

removed pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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, pipe
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():