Add types

README.md

@@ -1,7 +1,7 @@
 ---
 title: Chatbot
 emoji: 💬
-colorFrom: yellow
+colorFrom: blue
 colorTo: purple
 sdk: gradio
 app_file: app.py
@@ -9,4 +9,4 @@ pinned: false
 license: mit
 ---
 
-An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).
+An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).

app.py

@@ -1,19 +1,21 @@
+"""
+For more information on `huggingface_hub` Inference API support,
+please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
+"""
+
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
 def respond(
-    message,
+    message: str,
     history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
+    system_message: str,
+    max_tokens: int,
+    temperature: float,
+    top_p: float,
 ):
     messages = [{"role": "system", "content": system_message}]
 
@@ -39,9 +41,13 @@ def respond(
         response += token
         yield response
 
+
 """
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
+For information on how to customize the ChatInterface,
+peruse the gradio docs: https://www.gradio.app/docs/chatinterface
 """
+
+
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
@@ -60,4 +66,4 @@ demo = gr.ChatInterface(
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()