K00B404 commited on
Commit
0dc3b4b
1 Parent(s): 2e6bfa4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,8 +6,8 @@ import gradio as gr
6
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
 
8
 
9
- tokenizer = AutoTokenizer.from_pretrained("t5-base")
10
- model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
11
 
12
  def predict(input_text):
13
  inputs = tokenizer(input_text, return_tensors="pt")
@@ -20,7 +20,7 @@ iface = gr.Interface(fn=predict,
20
  placeholder="Enter your text..."),
21
  outputs="text",
22
  title="My First App",
23
- description="Summarize large blocks of text.",
24
  theme="dark",
25
  analytics_enabled=False)
26
 
 
6
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
 
8
 
9
+ tokenizer = AutoTokenizer.from_pretrained("cognitivecomputations/dolphin-2_6-phi-2t5-base")
10
+ model = AutoModelForSeq2SeqLM.from_pretrained("cognitivecomputations/dolphin-2_6-phi-2")
11
 
12
  def predict(input_text):
13
  inputs = tokenizer(input_text, return_tensors="pt")
 
20
  placeholder="Enter your text..."),
21
  outputs="text",
22
  title="My First App",
23
+ description="you are a good teacher of python code generation student models.",
24
  theme="dark",
25
  analytics_enabled=False)
26