Hugging Face's logo

Spaces:
jvillar02
/
news-classifier-demo
Running

App Files Community
Joaquin Villar commited on
Commit
271fe07
·
verified ·
1 Parent(s): 79f8da9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from peft import PeftModel
5
 
6
  # --- CONFIGURATION ---
7
  # Replace with your specific repo name
8
- ADAPTER_REPO = "jvillar-sheff/news-classifier-demo"
9
  BASE_MODEL_ID = "distilbert-base-uncased"
10
 
11
  CLASS_NAMES = {0: "World", 1: "Sports", 2: "Business", 3: "Sci/Tech"}
 
5
 
6
  # --- CONFIGURATION ---
7
  # Replace with your specific repo name
8
+ ADAPTER_REPO = "jvillar-sheff/ag-news-distilbert-lora"
9
  BASE_MODEL_ID = "distilbert-base-uncased"
10
 
11
  CLASS_NAMES = {0: "World", 1: "Sports", 2: "Business", 3: "Sci/Tech"}