ImHadis commited on
Commit
77b3501
·
verified ·
1 Parent(s): 4c66a5c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +35 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import io
3
+ from IPython.display import Image, display, HTML
4
+ from PIL import Image
5
+ import base64
6
+ from dotenv import load_dotenv, find_dotenv
7
+ import requests, json
8
+ from transformers import pipeline
9
+ import gradio as gr
10
+
11
+ # Access the Hugging Face API key and endpoint URL
12
+ hf_api_key = os.getenv('HF_API_KEY')
13
+ ENDPOINT_URL = os.getenv('HF_API_SUMMARY_BASE')
14
+
15
+ #Summarization endpoint
16
+ def get_completion(inputs, parameters=None,ENDPOINT_URL=os.environ['HF_API_SUMMARY_BASE']):
17
+ headers = {
18
+ "Authorization": f"Bearer {hf_api_key}",
19
+ "Content-Type": "application/json"
20
+ }
21
+ data = { "inputs": inputs }
22
+ if parameters is not None:
23
+ data.update({"parameters": parameters})
24
+ response = requests.request("POST",
25
+ ENDPOINT_URL, headers=headers,
26
+ data=json.dumps(data)
27
+ )
28
+ return json.loads(response.content.decode("utf-8"))
29
+
30
+ def summarize(input):
31
+ output = get_completion(input)
32
+ return output[0]['summary_text']
33
+
34
+ demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
35
+ demo.launch()