Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import pandas as pd
 
+from datasets import Dataset,DatasetDict
+
 df = pd.read_csv(path/'train.csv')
 
 df.describe(include='object')
@@ -9,12 +11,16 @@ df['input'] = 'TEXT1: ' + df.context + '; TEXT2: ' + df.target + '; ANC1: ' + df
 
 df.input.head()
 
-from datasets import Dataset,DatasetDict
 
 ds = Dataset.from_pandas(df)
 
-def greet(name):
-    return "Hello " + name + "!!"
+def predict_text(input_text):
+    prediction = classifier(input_text)
+    return prediction
+
+
+text_input = gr.inputs.Textbox(lines=7, label="Unesite tekst")
+output_text = gr.outputs.Textbox(label="Predikcija")
+
+gr.Interface(predict_text, inputs=text_input, outputs=output_text).launch()
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()