Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ int2label = {i: disease for i, disease in enumerate(df['label'].unique())}
|
|
| 13 |
label2int = {v: k for k, v in int2label.items()}
|
| 14 |
|
| 15 |
# Load the model and tokenizer
|
| 16 |
-
model_name = "
|
| 17 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 19 |
|
|
|
|
| 13 |
label2int = {v: k for k, v in int2label.items()}
|
| 14 |
|
| 15 |
# Load the model and tokenizer
|
| 16 |
+
model_name = "samyak152002/my-disease-classifier-sih" # Replace with your Hugging Face model path
|
| 17 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 19 |
|