Spaces:

Alesteba
/

AP_07

Sleeping

Alesteba commited on Apr 11, 2023

Commit

1c6650e

1 Parent(s): 04139f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,6 @@ classifier = pipeline("news-classifier", model="Alesteba/deep-model-07")
 # model = pipeline("Alesteba/deep-model-07")
 def predict(prompt):
-    completion = classifier(prompt)[0]["generated_text"]
-    return completion
 gr.Interface(fn=predict, inputs="text", outputs="text").launch()

 # model = pipeline("Alesteba/deep-model-07")
 def predict(prompt):
+    return classifier(prompt)
 gr.Interface(fn=predict, inputs="text", outputs="text").launch()