MMADS commited on
Commit
9c117f3
·
1 Parent(s): 465ebac

update model to gemma-3-270m-it

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -357,7 +357,7 @@ class CVEDashboard:
357
 
358
  def generate_tailored_summary(cve_description: str, audience: str, hf_token: Optional[str] = None, max_retries: int = 2) -> str:
359
  """
360
- Generates a tailored CVE summary using google/gemma-2b-it via HuggingFace Inference API.
361
 
362
  Args:
363
  cve_description: The original CVE description
@@ -382,7 +382,7 @@ def generate_tailored_summary(cve_description: str, audience: str, hf_token: Opt
382
 
383
  # Define the model to use
384
  models = [
385
- "google/gemma-2b-it",
386
  ]
387
 
388
  headers = {"Authorization": f"Bearer {token}"}
@@ -588,7 +588,7 @@ def create_interface():
588
  - Search CVEs by date range and keywords
589
  - Filter by severity levels
590
  - Visualize CVE distributions and trends
591
- - AI-powered audience-specific summaries using the SmolLM3-3B model.
592
 
593
  **Supported Audiences:**
594
  - **Cybersecurity Professional:** Focus on threats, attack vectors, and mitigation
@@ -600,9 +600,11 @@ def create_interface():
600
 
601
  **Data Source:** [NIST NVD API](https://nvd.nist.gov/developers/vulnerabilities)
602
 
603
- **AI Model:** [google/gemma-2b-it](https://huggingface.co/google/gemma-2b-it)
604
 
605
  **Disclaimer:** Generated content may be inaccurate or false.
 
 
606
 
607
  **Performance Optimizations:**
608
  - Shorter timeouts for faster failure detection
 
357
 
358
  def generate_tailored_summary(cve_description: str, audience: str, hf_token: Optional[str] = None, max_retries: int = 2) -> str:
359
  """
360
+ Generates a tailored CVE summary using google/gemma-3-270m-it via HuggingFace Inference API.
361
 
362
  Args:
363
  cve_description: The original CVE description
 
382
 
383
  # Define the model to use
384
  models = [
385
+ "google/gemma-3-270m-it",
386
  ]
387
 
388
  headers = {"Authorization": f"Bearer {token}"}
 
588
  - Search CVEs by date range and keywords
589
  - Filter by severity levels
590
  - Visualize CVE distributions and trends
591
+ - AI-powered audience-specific summaries using the google/gemma-3-270m-it model.
592
 
593
  **Supported Audiences:**
594
  - **Cybersecurity Professional:** Focus on threats, attack vectors, and mitigation
 
600
 
601
  **Data Source:** [NIST NVD API](https://nvd.nist.gov/developers/vulnerabilities)
602
 
603
+ **AI Model:** [google/gemma-3-270m-it](https://huggingface.co/google/gemma-3-270m-it)
604
 
605
  **Disclaimer:** Generated content may be inaccurate or false.
606
+
607
+ The free community tier of the Hugging Face Inference API powers this app's AI features. Since computing resources are shared, anticipate some delay on your initial request as the model loads. Later requests usually process more quickly.
608
 
609
  **Performance Optimizations:**
610
  - Shorter timeouts for faster failure detection