Gopikanth123 commited on
Commit
6db9ccd
·
verified ·
1 Parent(s): 938fd74

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +15 -2
main.py CHANGED
@@ -11,7 +11,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
11
  if not HF_TOKEN:
12
  raise ValueError("HF_TOKEN environment variable not set.")
13
 
14
- repo_id = "meta-llama/Meta-Llama-3-8B-Instruct"
15
  llm_client = InferenceClient(
16
  model=repo_id,
17
  token=HF_TOKEN,
@@ -29,6 +29,19 @@ llm_client = InferenceClient(
29
  # Settings.embed_model = HuggingFaceEmbedding(
30
  # model_name="BAAI/bge-small-en-v1.5"
31
  # )
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  # Configure Llama index settings
33
  Settings.llm = HuggingFaceInferenceAPI(
34
  model_name="xlm-roberta-base",
@@ -40,7 +53,7 @@ Settings.llm = HuggingFaceInferenceAPI(
40
  )
41
 
42
  Settings.embed_model = HuggingFaceEmbedding(
43
- model_name="sentence-transformers/paraphrase-xlm-r-100langs-v1"
44
  )
45
 
46
  PERSIST_DIR = "db"
 
11
  if not HF_TOKEN:
12
  raise ValueError("HF_TOKEN environment variable not set.")
13
 
14
+ repo_id = "xlm-roberta-base"
15
  llm_client = InferenceClient(
16
  model=repo_id,
17
  token=HF_TOKEN,
 
29
  # Settings.embed_model = HuggingFaceEmbedding(
30
  # model_name="BAAI/bge-small-en-v1.5"
31
  # )
32
+ # # Configure Llama index settings
33
+ # Settings.llm = HuggingFaceInferenceAPI(
34
+ # model_name="xlm-roberta-base",
35
+ # tokenizer_name="xlm-roberta-base",
36
+ # context_window=3000,
37
+ # token=HF_TOKEN,
38
+ # max_new_tokens=512,
39
+ # generate_kwargs={"temperature": 0.1},
40
+ # )
41
+
42
+ # Settings.embed_model = HuggingFaceEmbedding(
43
+ # model_name="sentence-transformers/paraphrase-xlm-r-100langs-v1"
44
+ # )
45
  # Configure Llama index settings
46
  Settings.llm = HuggingFaceInferenceAPI(
47
  model_name="xlm-roberta-base",
 
53
  )
54
 
55
  Settings.embed_model = HuggingFaceEmbedding(
56
+ model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2" # Updated model name
57
  )
58
 
59
  PERSIST_DIR = "db"