Update gradio_exit.py

gradio_exit.py

@@ -8,14 +8,12 @@ import base64
 from dotenv import load_dotenv, find_dotenv
 
 _ = load_dotenv(find_dotenv()) # read local .env file
-hf_api_key =os.getenv(hf_api_key)
 
-#model_id = "shleifer/distilbart-cnn-12-6"
-model_id = "flan-t5-base-text_summarization_data"
+model_id = os.getenv(model_id)
+hf_api_key = os.getenv(hf_api_key)
 
-api_url ='https://api-inference.huggingface.co/models/DunnBC22/{model_id}'
-
-headers = {"Authorization": f"Bearer {hf_api_key}"}
+#api_url ="https://api-inference.huggingface.co/models/DunnBC22/flan-t5-base-text_summarization_data"
+api_url =f"https://api-inference.huggingface.co/models/DunnBC22/{model_id}"
 
 def get_completion(inputs, parameters=None, ENDPOINT_URL=api_url): 
     headers = {
@@ -31,16 +29,7 @@ def get_completion(inputs, parameters=None, ENDPOINT_URL=api_url):
                                )
     return json.loads(response.content.decode("utf-8"))
 
-text = (
-    '''Alexander III of Macedon (Ancient Greek: Ἀλέξανδρος, romanized: Alexandros; 20/21 July 356 BC – 10/11 June 323 BC), '''
-    '''commonly known as Alexander the Great,[a] was a king of the ancient Greek kingdom of Macedon.[a] '''
-    '''He succeeded his father Philip II to the throne in 336 BC at the age of 20, '''
-    '''and spent most of his ruling years conducting a lengthy military campaign throughout Western Asia and Egypt. '''
-    '''By the age of 30, he had created one of the largest empires in history, stretching from Greece to northwestern India.[2] '''
-    '''He was undefeated in battle and is widely considered to be one of history's greatest and most successful military commanders.[3][4]'''
-)
-
-get_completion(text) 
+get_completion(text)
 
 import gradio as gr
 
@@ -53,7 +42,7 @@ gr.close_all()
 demo = gr.Interface(fn=summarize, 
                     inputs=[gr.Textbox(label="Text to summarize", lines=6)],
                     outputs=[gr.Textbox(label="Result", lines=3)],
-                    title="Text summarization with distilbart-cnn",
+                    title="基于HF requests.request的文本总结AI App/Text summarization with distilbart-cnn",
                     description="Summarize any text using the `shleifer/distilbart-cnn-12-6` model under the hood!"
                    )