Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import base64
|
|
| 2 |
from flask import Flask, render_template, request, jsonify, Response, stream_with_context
|
| 3 |
from huggingface_hub import login
|
| 4 |
from vismem.core import SectorMemory
|
| 5 |
-
from ai_engine import LocalModelHandler
|
| 6 |
|
| 7 |
import json
|
| 8 |
import os
|
|
@@ -51,7 +51,7 @@ def chat():
|
|
| 51 |
|
| 52 |
def generate():
|
| 53 |
# 1. VECTORIZE USER INPUT
|
| 54 |
-
q_vec =
|
| 55 |
|
| 56 |
# B. Relevant Past History
|
| 57 |
hist_hits = mem_history.search(q_vec, top_k=2)
|
|
|
|
| 2 |
from flask import Flask, render_template, request, jsonify, Response, stream_with_context
|
| 3 |
from huggingface_hub import login
|
| 4 |
from vismem.core import SectorMemory
|
| 5 |
+
from ai_engine import LocalModelHandler, get_embeddings
|
| 6 |
|
| 7 |
import json
|
| 8 |
import os
|
|
|
|
| 51 |
|
| 52 |
def generate():
|
| 53 |
# 1. VECTORIZE USER INPUT
|
| 54 |
+
q_vec = get_embedding(user_msg)
|
| 55 |
|
| 56 |
# B. Relevant Past History
|
| 57 |
hist_hits = mem_history.search(q_vec, top_k=2)
|