RoAr777 commited on
Commit
0564636
·
1 Parent(s): cfb525b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,6 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
  import time
3
  import gradio as gr
4
 
5
- model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-ul2")
6
- tokenizer = AutoTokenizer.from_pretrained("google/flan-ul2")
7
 
8
 
9
 
@@ -15,6 +13,8 @@ prompt=''' Review the Following text as a human, who is tasked to Extract `Code
15
  "{}"
16
  '''
17
  def reply(message, history):
 
 
18
  m=message.replace('\n','')
19
  m=m.replace('\t','')
20
  inputs = tokenizer(prompt.format(m), return_tensors="pt")
 
2
  import time
3
  import gradio as gr
4
 
 
 
5
 
6
 
7
 
 
13
  "{}"
14
  '''
15
  def reply(message, history):
16
+ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-ul2")
17
+ tokenizer = AutoTokenizer.from_pretrained("google/flan-ul2")
18
  m=message.replace('\n','')
19
  m=m.replace('\t','')
20
  inputs = tokenizer(prompt.format(m), return_tensors="pt")