ancebuc commited on
Commit
aabe7d9
·
verified ·
1 Parent(s): 78dd253

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,9 +8,9 @@ with gr.Blocks() as demo:
8
  msg = gr.Textbox()
9
  clear = gr.ClearButton([msg, chatbot])
10
 
11
- model = AutoModelForSequenceClassification.from_pretrained("/path/to/serialized/model/")
12
 
13
- tokenizer = AutoTokenizer.from_pretrained("/path/to/serialized/tokenizer/")
14
 
15
  query_pipeline = transformers.pipeline(
16
  "text-generation",
@@ -19,7 +19,7 @@ with gr.Blocks() as demo:
19
  torch_dtype=torch.float16,
20
  device_map="auto", max_new_tokens=200)
21
 
22
- vectordb = Chroma.from_documents(documents=all_splits, embedding=embeddings, persist_directory="chroma_db")
23
 
24
  def test_rag(pipeline, query):
25
  docs = vectordb.similarity_search_with_score(query)
 
8
  msg = gr.Textbox()
9
  clear = gr.ClearButton([msg, chatbot])
10
 
11
+ model = AutoModelForSequenceClassification.from_pretrained("ancebuc/scrum-chatbot/modelo")
12
 
13
+ tokenizer = AutoTokenizer.from_pretrained("ancebuc/scrum-chatbot/tokenizer")
14
 
15
  query_pipeline = transformers.pipeline(
16
  "text-generation",
 
19
  torch_dtype=torch.float16,
20
  device_map="auto", max_new_tokens=200)
21
 
22
+ vectordb = Chroma.load(persist_directory="ancebuc/scrum-chatbot/chroma")
23
 
24
  def test_rag(pipeline, query):
25
  docs = vectordb.similarity_search_with_score(query)