File size: 1,082 Bytes
9d9c274 e909703 9abed3a 18f5113 43dcf13 e085476 14f5c56 24668a8 4393529 18f5113 e909703 9abed3a e909703 9abed3a bf01bb4 d6af0b7 bf01bb4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from flask import Flask, jsonify, render_template, request, make_response
import requests
import transformers
from huggingface_hub import cached_download
import torch
from torch import nn
import re
import numpy as np
import pandas as pd
from collections import OrderedDict
# import requests
# from bs4 import BeautifulSoup
app = Flask(__name__)
headers = {"Authorization": f"Bearer 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"}
API_URL = "https://api-inference.huggingface.co/models/nlptown/bert-base-multilingual-uncased-sentiment"
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.json()
@app.route('/', methods=['GET'])
def get():
data = query({"inputs": "The movie is good"})
return data
@app.route("/", methods=["GET", "POST"])
def index():
user_input = None
response = None
if request.method == "POST":
user_input = request.form.get("user_input")
response = query({"inputs": user_input})
return render_template("home.html", user_input=user_input, response=response)
|