R-classify:从受控词汇表中提取研究论文的相关概念

0 下载量 30 浏览量 更新于2025-01-16 收藏 999KB PDF 举报

from flask import Flask, request, jsonify import torch from transformers import BertTokenizer, BertForSequenceClassification import logging app = Flask(name) logging.basicConfig(level=logging.INFO) tokenizer = BertTokenizer.from_pretrained('bert-base-chinese') model = BertForSequenceClassification.from_pretrained('bert-base-chinese', num_labels=2) model.eval() @app.route('/classify', methods=['POST']) def classify(): try: text = request.json['text'] inputs = tokenizer(text, return_tensors='pt') outputs = model(**inputs) logits = outputs.logits probabilities = torch.softmax(logits, dim=1) predicted_label = torch.argmax(probabilities, dim=1).item() if predicted_label == 0: result = '负面' else: result = '正面' logging.info(f'Text: {text}, Result: {result}') return jsonify({'result': result}) except Exception as e: logging.error(f'Error: {e}') return jsonify({'error': str(e)}) if name == 'main': app.run()这个是我的py代码,同一文件夹下,有一个html代码,叫做classify.html,代码如下:<!DOCTYPE html> <html> <head> <meta charset="UTF-8"> <title>BERT文本分类</title> <style> body { font-family: Arial, sans-serif; } h1 { text-align: center; } form { margin: 0 auto; width: 50%; text-align: center; } label { display: block; margin-top: 20px; } input[type="text"] { width: 100%; padding: 10px; border: 1px solid #ccc; border-radius: 4px; box-sizing: border-box; } input[type="submit"] { background-color: #4CAF50; color: white; padding: 10px 20px; border: none; border-radius: 4px; cursor: pointer; } input[type="submit"]:hover { background-color: #45a049; } #result { margin-top: 20px; padding: 10px; border: 1px solid #ccc; border-radius: 4px; background-color: #f2f2f2; } </style> </head> <body>

BERT文本分类

<form> <label for="text">请输入文本:</label> <input type="text" id="text" name="text"> <input type="submit" value="提交"> </form>
<script> const form = document.querySelector('form'); const resultDiv = document.querySelector('#result'); form.addEventListener('submit', (event) => { event.preventDefault(); const text = document.querySelector('#text').value; fetch('/classify', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({text: text}) }) .then(response => response.json()) .then(data => { resultDiv.innerHTML =

分类结果:${data.result}

; }) .catch(error => { resultDiv.innerHTML =

出错了:${error.message}

; }); }); </script> </body> </html>,请问为什么打开后有问题呢,帮我解决一下,优化下代码
183 浏览量