Added model

.vscode/settings.json

@@ -0,0 +1,6 @@
+{
+    "[python]": {
+        "editor.defaultFormatter": "ms-python.black-formatter"
+    },
+    "python.formatting.provider": "none"
+}

app.py

@@ -1,7 +1,18 @@
 import gradio as gr
+import torch
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+generate_text = pipeline(
+    model="databricks/dolly-v2-12b",
+    torch_dtype=torch.bfloat16,
+    trust_remote_code=True,
+    device_map="auto",
+)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+def text_generation(prompt: str):
+    return generate_text(prompt)
+
+
+iface = gr.Interface(fn=text_generation, inputs="text", outputs="text")
+iface.launch()