Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("/
|
12 |
|
13 |
-
tokenizer = AutoTokenizer.from_pretrained("/
|
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.
|
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)
|