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)
|