Ahmed commited on
Commit
ecc129a
·
verified ·
1 Parent(s): 90a4ac7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,11 +6,11 @@ instruct_model_name='truocpham/flan-dialogue-summary-checkpoint'
6
  tokenizer = AutoTokenizer.from_pretrained(tokenizerModelName)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(instruct_model_name)
8
 
9
- def SummarizeThis(dialogue):
10
  prompt = f"""
11
  Summarize the following conversation.
12
 
13
- {dialogue}
14
 
15
  Summary:
16
  """
 
6
  tokenizer = AutoTokenizer.from_pretrained(tokenizerModelName)
7
  model = AutoModelForSeq2SeqLM.from_pretrained(instruct_model_name)
8
 
9
+ def SummarizeThis(Dialogue):
10
  prompt = f"""
11
  Summarize the following conversation.
12
 
13
+ {Dialogue}
14
 
15
  Summary:
16
  """