Create app.py

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from transformers import AutoTokenizer, FalconModel
+
+tokenizer = AutoTokenizer.from_pretrained("Rocketknight1/falcon-rw-1b")
+model = FalconModel.from_pretrained("Rocketknight1/falcon-rw-1b")
+
+def chat(input_text):
+    inputs = tokenizer(input_text, return_tensors="pt")
+    outputs = model(**inputs)
+    last_hidden_states = outputs.last_hidden_state
+    response = tokenizer.decode(last_hidden_states.squeeze(), skip_special_tokens=True)
+    return response
+
+interface = gr.Interface(
+    fn=chat,
+    inputs=gr.inputs.Textbox(lines=2, label="User Question"),
+    outputs=gr.outputs.Textbox(label="Chatbot Response"),
+    title="Falcon-7b",
+    theme="dark"
+)
+
+interface.launch()