0708_7

app.py

@@ -15,12 +15,12 @@ print("downloading model...")
 sen_model_file = hf_hub_download(repo_id=SENTIMENT_REPO_ID, filename=SENTIMENT_FILENAME)
 entity_model_file = hf_hub_download(repo_id=ENTITY_REPO_ID, filename=ENTITY_FILENAME)
 
-base_model = cfg.base_model
+base_model = base_model
 
 tokenizer = AutoTokenizer.from_pretrained(base_model)
 
 sen_model = Classifier(base_model, num_labels=2, device=device, tokenizer=tokenizer)
-sen_model.load_state_dict(torch.load(sen_model_file))
+sen_model.load_state_dict(torch.load('/resolve/main/'+sen_model_file))
 
 entity_model = Classifier(base_model, num_labels=2, device=device, tokenizer=tokenizer)
 entity_model.load_state_dict(torch.load(entity_model_file))
@@ -75,7 +75,7 @@ def infer(test_sentence):
     result = '\n'.join(annotation)
     return result
 
-article = "**이미지를 업로드하세요.**" \
+article = "**분석할 텍스트를 입력하세요.**" \
     
 demo = gr.Interface(fn=infer,
              inputs=gr.Textbox(type="text", label="Input Sentence"),