Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ import gradio as gr
|
|
6 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
7 |
|
8 |
|
9 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
10 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
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="
|
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 |
|