Spaces:
Runtime error
Runtime error
| import os | |
| import requests, json | |
| from dotenv import load_dotenv, find_dotenv #pip install python-dotenv | |
| load_dotenv(find_dotenv()) # read local .env file | |
| hf_api_key = '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' | |
| #Summarization endpoint | |
| def get_completion(inputs, parameters=None,ENDPOINT_URL="https://api-inference.huggingface.co/models/sshleifer/distilbart-cnn-12-6"): | |
| headers = { | |
| "Authorization": f"Bearer {hf_api_key}", | |
| "Content-Type": "application/json" | |
| } | |
| data = { "inputs": inputs } | |
| if parameters is not None: | |
| data.update({"parameters": parameters}) | |
| response = requests.request("POST", | |
| ENDPOINT_URL, headers=headers, | |
| data=json.dumps(data) | |
| ) | |
| return json.loads(response.content.decode("utf-8")) | |
| text = ('''The tower is 324 metres (1,063 ft) tall, about the same height | |
| as an 81-storey building, and the tallest structure in Paris. | |
| Its base is square, measuring 125 metres (410 ft) on each side. | |
| During its construction, the Eiffel Tower surpassed the Washington | |
| Monument to become the tallest man-made structure in the world, | |
| a title it held for 41 years until the Chrysler Building | |
| in New York City was finished in 1930. It was the first structure | |
| to reach a height of 300 metres. Due to the addition of a broadcasting | |
| aerial at the top of the tower in 1957, it is now taller than the | |
| Chrysler Building by 5.2 metres (17 ft). Excluding transmitters, the | |
| Eiffel Tower is the second tallest free-standing structure in France | |
| after the Millau Viaduct.''') | |
| result = get_completion(text) | |
| print(result) | |
| import gradio as gr | |
| def summarize(input): | |
| output = get_completion(input) | |
| return output[0]['summary_text'] | |
| demo = gr.Interface(fn=summarize, inputs="text", outputs="text") | |
| demo.launch() |