Update app.py

app.py

@@ -1,8 +1,20 @@
+# import gradio as gr
+# from transformers import pipeline
+# model = pipeline("text-generation", model="DmitryYarov/plato_2024_1.0")
+# def predict(prompt):
+#     completion = model(prompt)[0]["generated_text"]
+#     return completion
+
+# gr.Interface(fn=predict, inputs="text", outputs="text").launch()
 import gradio as gr
-from transformers import pipeline
-model = pipeline("text-generation", model="DmitryYarov/plato_2024_1.0")
-def predict(prompt):
-    completion = model(prompt)[0]["generated_text"]
-    return completion
 
-gr.Interface(fn=predict, inputs="text", outputs="text").launch()
+def greet(name, intensity):
+    return "Hello, " + name + "!" * int(intensity)
+
+demo = gr.Interface(
+    fn=greet,
+    inputs=["text", "slider"],
+    outputs=["text"],
+)
+
+demo.launch()