RoAr777 commited on
Commit
1f9515f
·
1 Parent(s): f067917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import time
3
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
- prompt=''' Review the Following text as a human, who is tasked to EXTRACT Code Snippets ,if any otherwise return No:
5
 
6
  "/{}/"
7
  '''
@@ -11,8 +11,8 @@ replacements = {
11
  '<': 'LessThanSign'
12
  }
13
 
14
- model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
15
- tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
16
 
17
  def chatbot_model(m,history):
18
  for char, replacement in replacements.items():
 
1
  import gradio as gr
2
  import time
3
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
+ prompt=''' Review the Following text as a human, who is tasked to Extract `Code Snippets` :
5
 
6
  "/{}/"
7
  '''
 
11
  '<': 'LessThanSign'
12
  }
13
 
14
+ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-ul2")
15
+ tokenizer = AutoTokenizer.from_pretrained("google/flan-ul2")
16
 
17
  def chatbot_model(m,history):
18
  for char, replacement in replacements.items():