Learto commited on
Commit
ed3ffd7
·
1 Parent(s): 2306ef5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import requests
2
- import torch
3
  import gradio as gr
4
 
5
  API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
@@ -7,7 +7,7 @@ headers = {"Authorization": "Bearer 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"}
7
 
8
  def query(payload):
9
  response = requests.post(API_URL, headers=headers, json=payload)
10
- return response.items()
11
 
12
  output = query({
13
  "inputs": "I like you. I love you",
 
1
  import requests
2
+ import urllib.parse
3
  import gradio as gr
4
 
5
  API_URL = "https://api-inference.huggingface.co/models/ProsusAI/finbert"
 
7
 
8
  def query(payload):
9
  response = requests.post(API_URL, headers=headers, json=payload)
10
+ return response.json()['label']
11
 
12
  output = query({
13
  "inputs": "I like you. I love you",