Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,10 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
from googletrans import Translator
|
4 |
import requests
|
5 |
from dotenv import load_dotenv
|
6 |
import numpy as np
|
7 |
-
from langchain_community.embeddings import HuggingFaceEmbeddings
|
8 |
from langchain_community.vectorstores import Chroma
|
9 |
from langchain_community.document_loaders import UnstructuredPDFLoader
|
10 |
from langchain.text_splitter import CharacterTextSplitter
|
@@ -23,6 +23,10 @@ import logging
|
|
23 |
# Aggiornamento dell'inizializzazione di HuggingFaceEmbeddings
|
24 |
embedding_function = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
25 |
|
|
|
|
|
|
|
|
|
26 |
logging.basicConfig(level=logging.INFO)
|
27 |
logger = logging.getLogger(__name__)
|
28 |
|
|
|
1 |
+
from langchain_huggingface import HuggingFaceEmbeddings
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
from googletrans import Translator
|
5 |
import requests
|
6 |
from dotenv import load_dotenv
|
7 |
import numpy as np
|
|
|
8 |
from langchain_community.vectorstores import Chroma
|
9 |
from langchain_community.document_loaders import UnstructuredPDFLoader
|
10 |
from langchain.text_splitter import CharacterTextSplitter
|
|
|
23 |
# Aggiornamento dell'inizializzazione di HuggingFaceEmbeddings
|
24 |
embedding_function = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
25 |
|
26 |
+
# Definizione della lista di modelli LLM
|
27 |
+
list_llm_simple = ["Gemma 7B (Italian)", "Mistral 7B"]
|
28 |
+
list_llm = ["google/gemma-7b-it", "mistralai/Mistral-7B-Instruct-v0.2"]
|
29 |
+
|
30 |
logging.basicConfig(level=logging.INFO)
|
31 |
logger = logging.getLogger(__name__)
|
32 |
|