Michael Brunzel commited on
Commit
e8c7621
·
1 Parent(s): 6619948

use regular requests package

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -1,24 +1,24 @@
1
  import gradio as gr
2
  import os
 
3
  from dotenv import load_dotenv
4
- from text_generation import Client
5
 
6
  load_dotenv()
7
 
8
  API_TOKEN = os.environ.get("API_TOKEN", None)
9
  MODEL_URL = os.environ.get("MODEL_URL", None)
10
 
11
- def evaluate(hotel_request: str, **kwargs):
12
- hf_client = Client(
13
  MODEL_URL,
 
14
  headers={"Authorization": f"Bearer {API_TOKEN}"},
 
 
15
  )
 
16
 
17
- output = hf_client.generate(
18
- hotel_request,
19
- **kwargs,
20
- )
21
- text = output.generated_text
22
  name, location, hotel, date = text.split("|")
23
 
24
  return name, hotel, location, date
 
1
  import gradio as gr
2
  import os
3
+ import requests
4
  from dotenv import load_dotenv
 
5
 
6
  load_dotenv()
7
 
8
  API_TOKEN = os.environ.get("API_TOKEN", None)
9
  MODEL_URL = os.environ.get("MODEL_URL", None)
10
 
11
+ def evaluate(hotel_request: str):
12
+ resp = requests.post(
13
  MODEL_URL,
14
+ json={"inputs": hotel_request},
15
  headers={"Authorization": f"Bearer {API_TOKEN}"},
16
+ cookies=None,
17
+ timeout=10,
18
  )
19
+ payload = resp.json()
20
 
21
+ text = payload[0]["generated_text"]
 
 
 
 
22
  name, location, hotel, date = text.split("|")
23
 
24
  return name, hotel, location, date