pythonprincess commited on
Commit
bde56f5
·
verified ·
1 Parent(s): e6f71a9

Upload 2 files

Browse files
Files changed (2) hide show
  1. models/__init__.py +14 -0
  2. models/model_config.json +47 -0
models/__init__.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # PENNY Models Package
2
+ """
3
+ PENNY Model Utilities Package
4
+
5
+ This package contains specialized model utilities for:
6
+ - Translation (NLLB-200)
7
+ - Sentiment Analysis
8
+ - Bias Detection
9
+ - Document Processing (LayoutLM)
10
+ - Conversational AI (Gemma)
11
+ """
12
+
13
+ __version__ = "1.0.0"
14
+
models/model_config.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "penny-core-agent": {
3
+ "model_name": "google/gemma-7b-it",
4
+ "task": "text-generation",
5
+ "endpoint": "huggingface-api",
6
+ "api_url": "https://api-inference.huggingface.co/models/google/gemma-7b-it",
7
+ "timeout_seconds": 30,
8
+ "max_retries": 2,
9
+ "description": "Penny's core conversational AI for civic engagement responses"
10
+ },
11
+ "penny-doc-agent": {
12
+ "model_name": "microsoft/layoutlmv3-base",
13
+ "task": "pdf-extraction",
14
+ "endpoint": "huggingface-api",
15
+ "api_url": "https://api-inference.huggingface.co/models/microsoft/layoutlmv3-base",
16
+ "timeout_seconds": 45,
17
+ "max_retries": 2,
18
+ "description": "Document understanding and PDF extraction for civic documents"
19
+ },
20
+ "penny-translate-agent": {
21
+ "model_name": "facebook/nllb-200-distilled-600M",
22
+ "task": "translation",
23
+ "endpoint": "huggingface-api",
24
+ "api_url": "https://api-inference.huggingface.co/models/facebook/nllb-200-distilled-600M",
25
+ "timeout_seconds": 20,
26
+ "max_retries": 2,
27
+ "description": "Multilingual translation service for accessible civic information"
28
+ },
29
+ "penny-sentiment-agent": {
30
+ "model_name": "cardiffnlp/twitter-roberta-base-sentiment",
31
+ "task": "sentiment-analysis",
32
+ "endpoint": "huggingface-api",
33
+ "api_url": "https://api-inference.huggingface.co/models/cardiffnlp/twitter-roberta-base-sentiment",
34
+ "timeout_seconds": 15,
35
+ "max_retries": 2,
36
+ "description": "Sentiment analysis for community feedback and engagement monitoring"
37
+ },
38
+ "penny-bias-checker": {
39
+ "model_name": "facebook/bart-large-mnli",
40
+ "task": "bias-detection",
41
+ "endpoint": "huggingface-api",
42
+ "api_url": "https://api-inference.huggingface.co/models/facebook/bart-large-mnli",
43
+ "timeout_seconds": 20,
44
+ "max_retries": 2,
45
+ "description": "Bias detection to ensure fair and equitable civic information"
46
+ }
47
+ }