Hugging Face's logo

Spaces:
DIVY118
/
text-classifier
Runtime error

App Files Community
DIVY118 commited on
Commit
fd87d06
•
1 Parent(s): 532ddba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -11
app.py CHANGED
@@ -15,17 +15,23 @@ generation_config = {
15
  }
16
 
17
  safety_settings = [
18
- {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
19
- {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
20
- {
21
- "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
22
- "threshold": "BLOCK_MEDIUM_AND_ABOVE",
23
- },
24
- {
25
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
26
- "threshold": "BLOCK_MEDIUM_AND_ABOVE",
27
- },
28
- ]
 
 
 
 
 
 
29
 
30
  model = genai.GenerativeModel(
31
  model_name="gemini-pro",
 
15
  }
16
 
17
  safety_settings = [
18
+ {
19
+ "category": "HARM_CATEGORY_HARASSMENT",
20
+ "threshold": "BLOCK_NONE"
21
+ },
22
+ {
23
+ "category": "HARM_CATEGORY_HATE_SPEECH",
24
+ "threshold": "BLOCK_NONE"
25
+ },
26
+ {
27
+ "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
28
+ "threshold": "BLOCK_NONE"
29
+ },
30
+ {
31
+ "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
32
+ "threshold": "BLOCK_NONE"
33
+ },
34
+ ]
35
 
36
  model = genai.GenerativeModel(
37
  model_name="gemini-pro",