Spaces:

awacke1
/

sileod-deberta-v3-base-tasksource-nli

Runtime error

awacke1 commited on Feb 15, 2023

Commit

b886a66

1 Parent(s): 9d8750a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import gradio as gr
 def classify_text(text1, text2):
     # Load pre-trained model
-    model = gr.load_model("models/sileod/deberta-v3-base-tasksource-nli")
     # Perform classification on input text
     output = model.predict([text1, text2])[0]

 def classify_text(text1, text2):
     # Load pre-trained model
+    model = gr.Interface.load("models/sileod/deberta-v3-base-tasksource-nli")
     # Perform classification on input text
     output = model.predict([text1, text2])[0]