flaskapp / main.py
kumar9's picture
Update main.py
bf01bb4
raw
history blame
No virus
1.08 kB
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("index.html", user_input=user_input, response=response)