jknebel commited on
Commit
0d3fc53
·
1 Parent(s): 1dbc1fe

correction

Browse files
Files changed (1) hide show
  1. app.py +0 -11
app.py CHANGED
@@ -11,20 +11,9 @@ def predict(prompt,model_name, max_length):
11
  model_name = "google/pegasus-xsum"
12
  else:
13
  model_name = model_name_converter[model_name]
14
- print('la')
15
- print(model_name)
16
- print(max_length)
17
  model = pipeline("summarization",model = model_name)
18
  summary = model(prompt,max_length)[0]["summary_text"]
19
  return summary
20
-
21
- def extract_model(option):
22
- if option ==None:
23
- model_name = "google/pegasus-xsum"
24
- else:
25
- model_name = model_name_converter[option]
26
-
27
- return print(model_name)
28
 
29
  options_1 = model_name_converter.keys()
30
  with gr.Blocks() as demo:
 
11
  model_name = "google/pegasus-xsum"
12
  else:
13
  model_name = model_name_converter[model_name]
 
 
 
14
  model = pipeline("summarization",model = model_name)
15
  summary = model(prompt,max_length)[0]["summary_text"]
16
  return summary
 
 
 
 
 
 
 
 
17
 
18
  options_1 = model_name_converter.keys()
19
  with gr.Blocks() as demo: