ffreemt commited on
Commit
28743d9
1 Parent(s): f1dfff2

Update some fixes

Browse files
Files changed (2) hide show
  1. app.py +12 -7
  2. okteto.yml +2 -2
app.py CHANGED
@@ -236,8 +236,7 @@ def ingest(
236
  for doc in documents
237
  ]
238
 
239
-
240
- # TheBloke/vicuna-7B-1.1-GPTQ-4bit-128g
241
  def gen_local_llm(model_id="TheBloke/vicuna-7B-1.1-HF"):
242
  """Gen a local llm.
243
 
@@ -334,13 +333,19 @@ def main():
334
  def respond(message, chat_history):
335
  # bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
336
  if ns.qa is None: # no files processed yet
337
- return "Provide some file(s) for processsing first.", chat_history
 
 
 
 
 
 
 
 
 
338
 
339
- res = ns.qa(message)
340
- answer, docs = res["result"], res["source_documents"]
341
- bot_message = f"{answer} ({docs})"
342
  chat_history.append((message, bot_message))
343
- time.sleep(0.21)
344
  return "", chat_history
345
 
346
  msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
236
  for doc in documents
237
  ]
238
 
239
+ # https://huggingface.co/TheBloke/vicuna-7B-1.1-HF
 
240
  def gen_local_llm(model_id="TheBloke/vicuna-7B-1.1-HF"):
241
  """Gen a local llm.
242
 
 
333
  def respond(message, chat_history):
334
  # bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
335
  if ns.qa is None: # no files processed yet
336
+ bot_message = "Provide some file(s) for processsing first."
337
+ chat_history.append((message, bot_message))
338
+ return "", chat_history
339
+ try:
340
+ res = ns.qa(message)
341
+ answer, docs = res["result"], res["source_documents"]
342
+ bot_message = f"{answer} ({docs})"
343
+ except Exception as exc:
344
+ logger.error(exc)
345
+ bot_message = f"{exc}"
346
 
 
 
 
347
  chat_history.append((message, bot_message))
348
+
349
  return "", chat_history
350
 
351
  msg.submit(respond, [msg, chatbot], [msg, chatbot])
okteto.yml CHANGED
@@ -36,8 +36,8 @@ dev:
36
  environment:
37
  - name=$USER
38
  forward:
39
- - 7861:7861
40
- # - 7860:7860
41
  - 8501:8501
42
  - 22:22
43
  reverse:
 
36
  environment:
37
  - name=$USER
38
  forward:
39
+ # - 7861:7861
40
+ - 7860:7860
41
  - 8501:8501
42
  - 22:22
43
  reverse: