Kevin Fink
commited on
Commit
·
93bb0c2
1
Parent(s):
e4621e6
dev
Browse files
app.py
CHANGED
|
@@ -144,7 +144,7 @@ def predict(text):
|
|
| 144 |
'''
|
| 145 |
# Create Gradio interface
|
| 146 |
try:
|
| 147 |
-
|
| 148 |
iface = gr.Interface(
|
| 149 |
fn=fine_tune_model,
|
| 150 |
inputs=[
|
|
|
|
| 144 |
'''
|
| 145 |
# Create Gradio interface
|
| 146 |
try:
|
| 147 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2, force_download=True)
|
| 148 |
iface = gr.Interface(
|
| 149 |
fn=fine_tune_model,
|
| 150 |
inputs=[
|