1

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import torch
-from transformers import BertTokenizer, BertModel  # Pastikan ini ada
+from transformers import BertTokenizer, BertModel  
 import torch.nn.functional as F
 
 # Load model dan tokenizer dari Hugging Face
-model_name = "shobrunjb/mtl-indobert-fake-review-product"
+model_name = "shobrunjb/mtl-indoBERT-product-review"
 tokenizer = BertTokenizer.from_pretrained(model_name)
 
 class IndoBERTMultiTaskClassifier(torch.nn.Module):
@@ -27,7 +27,7 @@ class IndoBERTMultiTaskClassifier(torch.nn.Module):
 
 # Load the model
 model = IndoBERTMultiTaskClassifier(
-    bert_model_name="indobenchmark/indobert-base-p1",
+    bert_model_name="shobrunjb/mtl-indoBERT-product-review",
     num_labels_task1=3,  # Adjust with your task1 classes
     num_labels_task2=3   # Adjust with your task2 classes
 )