Spaces:

shorecode
/

gradio-3

Running

Kevin Fink commited on Dec 6, 2024

Commit

3ae598a

1 Parent(s): 93bb0c2

dev

Files changed (1) hide show

app.py CHANGED Viewed

@@ -144,7 +144,7 @@ def predict(text):
 '''
 # Create Gradio interface
 try:
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2, force_download=True)
     iface = gr.Interface(
         fn=fine_tune_model,
         inputs=[

 '''
 # Create Gradio interface
 try:
+    model = AutoModelForSeq2SeqLM.from_pretrained('google/t5-efficient-tiny-nh8'.strip(), num_labels=2, force_download=True)
     iface = gr.Interface(
         fn=fine_tune_model,
         inputs=[