farmax commited on
Commit
6739a2f
·
verified ·
1 Parent(s): eb90152

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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