Docfile commited on
Commit
a25995c
·
verified ·
1 Parent(s): c3a1a37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -1
app.py CHANGED
@@ -37,7 +37,30 @@ llm= LLM = G4FLLM(
37
  llm = LangChainLLM(llm=llm)
38
 
39
  """
40
- llm=ChatGoogleGenerativeAI(model="gemini-pro", convert_system_message_to_human=True )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
41
  model_name = "models/embedding-001"
42
 
43
 
@@ -62,6 +85,7 @@ documents = SimpleDirectoryReader("./data", file_extractor=file_extractor).load_
62
  """
63
 
64
 
 
65
  documents = (
66
  SimpleDirectoryReader(
67
  input_dir = 'data',
 
37
  llm = LangChainLLM(llm=llm)
38
 
39
  """
40
+
41
+ safe = [
42
+ {
43
+ "category": "HARM_CATEGORY_HARASSMENT",
44
+ "threshold": "BLOCK_NONE",
45
+ },
46
+ {
47
+ "category": "HARM_CATEGORY_HATE_SPEECH",
48
+ "threshold": "BLOCK_NONE",
49
+ },
50
+ {
51
+ "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
52
+ "threshold": "BLOCK_NONE",
53
+ },
54
+ {
55
+ "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
56
+ "threshold": "BLOCK_NONE",
57
+ },
58
+ ]
59
+
60
+
61
+
62
+ llm = Gemini(model="models/gemini-pro", safety_settings=safe)
63
+
64
  model_name = "models/embedding-001"
65
 
66
 
 
85
  """
86
 
87
 
88
+
89
  documents = (
90
  SimpleDirectoryReader(
91
  input_dir = 'data',