manojpatil commited on
Commit
847240a
·
1 Parent(s): 611a54e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -63,6 +63,7 @@ items = os.listdir(BASE_DIR)
63
  for item in items:
64
  print(item)
65
  #intialize replicate llm
 
66
  llm = Replicate(
67
  model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
68
  model_kwargs={"temperature": 0.1,
@@ -87,11 +88,11 @@ template_replicate = get_prompt(instruction_replicate,DEFAULT_SYSTEM_PROMPT_repl
87
 
88
  prompt_replicate = PromptTemplate(template=template_replicate,input_variables=['text'])
89
  llm_chain_Replicate = LLMChain(prompt=prompt_replicate, llm=llm)
90
-
91
  def llama2(query):
92
  try:
93
  text=query
94
- output = llm_chain_Replicate.run(text)
95
  except:
96
  pass
97
  return output
 
63
  for item in items:
64
  print(item)
65
  #intialize replicate llm
66
+ '''
67
  llm = Replicate(
68
  model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
69
  model_kwargs={"temperature": 0.1,
 
88
 
89
  prompt_replicate = PromptTemplate(template=template_replicate,input_variables=['text'])
90
  llm_chain_Replicate = LLMChain(prompt=prompt_replicate, llm=llm)
91
+ '''
92
  def llama2(query):
93
  try:
94
  text=query
95
+ #output = llm_chain_Replicate.run(text)
96
  except:
97
  pass
98
  return output