init

README.md

@@ -1,10 +1,10 @@
 ---
-title: Dolly Chat
-emoji: 🐠
-colorFrom: blue
-colorTo: gray
+title: Dolly
+emoji: 🌍
+colorFrom: gray
+colorTo: indigo
 sdk: gradio
-sdk_version: 3.34.0
+sdk_version: 3.29.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,9 @@
+import gradio as gr
+import torch
+from transformers import pipeline
+ans = pipeline(model="databricks/dolly-v2-3b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
+def answer(query):
+  out=ans(query)
+  return out
+Demo = gr.Interface(fn=answer,inputs='text',outputs='text',examples=[['What is the capital of India ?']])
+Demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch 
+accelerate