Spaces:

baadror
/

text_generation

Sleeping

baadror commited on Jun 21, 2024

Commit

331dc69

1 Parent(s): f72dedb

check

Files changed (1) hide show

main.py CHANGED Viewed

@@ -5,9 +5,9 @@ import gradio as gr
 # pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
-def pred(input):
-    output = pipe_flan(input)
-    return output[0]["generated_text"]
 def test_fun(input):
     return input

 # pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+# def pred(input):
+#     output = pipe_flan(input)
+#     return output[0]["generated_text"]
 def test_fun(input):
     return input