File size: 1,078 Bytes
9d9c274 e909703 9abed3a 18f5113 43dcf13 e085476 14f5c56 a793420 24668a8 4393529 a793420 18f5113 6013fef 5d683ae a793420 6013fef a793420 5d683ae e909703 6013fef 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 41 42 |
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 os
# import requests
# from bs4 import BeautifulSoup
app = Flask(__name__)
import os
# os.environ['API_URL'] = 'https://api-inference.huggingface.co/models/ProsusAI/finbert'
# API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
api_key = os.getenv('API_KEY')
api_url = os.getenv('API_URL')
headers = {"Authorization": f"Bearer {api_key}"}
def query(payload):
response = requests.post(api_url, headers=headers, json=payload)
return response.json()
@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)
|