Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ import numpy as np
|
|
| 9 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, BitsAndBytesConfig
|
| 10 |
from datetime import datetime
|
| 11 |
import json
|
|
|
|
| 12 |
|
| 13 |
class DocumentRetrievalAndGeneration:
|
| 14 |
def __init__(self, embedding_model_name, lm_model_id, data_folder, faiss_index_path):
|
|
|
|
| 9 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, BitsAndBytesConfig
|
| 10 |
from datetime import datetime
|
| 11 |
import json
|
| 12 |
+
import gradio as gr
|
| 13 |
|
| 14 |
class DocumentRetrievalAndGeneration:
|
| 15 |
def __init__(self, embedding_model_name, lm_model_id, data_folder, faiss_index_path):
|