Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ def local_query(query, context):
|
|
41 |
|
42 |
print('entering model generator')
|
43 |
outputs = model.generate(**inputs, max_new_tokens=20)
|
44 |
-
print(f'output is outputs')
|
45 |
return tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
46 |
|
47 |
def run_query(file, history, query):
|
@@ -70,8 +70,7 @@ def run_query(file, history, query):
|
|
70 |
ids=ids
|
71 |
)
|
72 |
|
73 |
-
|
74 |
-
print(collection)
|
75 |
|
76 |
context = get_context(query, collection)
|
77 |
|
|
|
41 |
|
42 |
print('entering model generator')
|
43 |
outputs = model.generate(**inputs, max_new_tokens=20)
|
44 |
+
print(f'output is {outputs}')
|
45 |
return tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
46 |
|
47 |
def run_query(file, history, query):
|
|
|
70 |
ids=ids
|
71 |
)
|
72 |
|
73 |
+
|
|
|
74 |
|
75 |
context = get_context(query, collection)
|
76 |
|