Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,30 @@ llm= LLM = G4FLLM(
|
|
37 |
llm = LangChainLLM(llm=llm)
|
38 |
|
39 |
"""
|
40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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',
|