Walid-Ahmed commited on
Commit
cadff73
·
verified ·
1 Parent(s): dcfe9dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -18,6 +18,15 @@ summarizer = None
18
  tokenizer = None
19
  max_tokens = None
20
 
 
 
 
 
 
 
 
 
 
21
 
22
  # Function to load the selected model
23
  def load_model(model_name):
@@ -35,7 +44,6 @@ def load_model(model_name):
35
  except Exception as e:
36
  return f"Failed to load model {model_name}. Error: {str(e)}"
37
 
38
-
39
  # Function to summarize the input text
40
  def summarize_text(input, min_length, max_length):
41
  if summarizer is None:
@@ -58,7 +66,6 @@ def summarize_text(input, min_length, max_length):
58
  except Exception as e:
59
  return f"Summarization failed: {str(e)}"
60
 
61
-
62
  # Gradio Interface
63
  with gr.Blocks() as demo:
64
  with gr.Row():
@@ -70,7 +77,7 @@ with gr.Blocks() as demo:
70
  min_length_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Minimum Summary Length (%)", value=10)
71
  max_length_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Maximum Summary Length (%)", value=20)
72
 
73
- input_text = gr.Textbox(label="Input text to summarize", lines=6)
74
  summarize_button = gr.Button("Summarize Text")
75
  output_text = gr.Textbox(label="Summarized text", lines=4)
76
 
 
18
  tokenizer = None
19
  max_tokens = None
20
 
21
+ # Example text for summarization
22
+ example_text = (
23
+ "Artificial intelligence (AI) is intelligence—perceiving, synthesizing, and inferring information—"
24
+ "demonstrated by machines, as opposed to intelligence displayed by non-human animals and humans. "
25
+ "Example tasks in which AI is employed include speech recognition, computer vision, language translation, "
26
+ "autonomous vehicles, and game playing. AI research has been defined as the field of study of intelligent "
27
+ "agents, which refers to any system that perceives its environment and takes actions that maximize its "
28
+ "chance of achieving its goals."
29
+ )
30
 
31
  # Function to load the selected model
32
  def load_model(model_name):
 
44
  except Exception as e:
45
  return f"Failed to load model {model_name}. Error: {str(e)}"
46
 
 
47
  # Function to summarize the input text
48
  def summarize_text(input, min_length, max_length):
49
  if summarizer is None:
 
66
  except Exception as e:
67
  return f"Summarization failed: {str(e)}"
68
 
 
69
  # Gradio Interface
70
  with gr.Blocks() as demo:
71
  with gr.Row():
 
77
  min_length_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Minimum Summary Length (%)", value=10)
78
  max_length_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Maximum Summary Length (%)", value=20)
79
 
80
+ input_text = gr.Textbox(label="Input text to summarize", lines=6, value=example_text)
81
  summarize_button = gr.Button("Summarize Text")
82
  output_text = gr.Textbox(label="Summarized text", lines=4)
83