Update Gradio app with multiple files

README.md

@@ -7,8 +7,6 @@ sdk: gradio
 sdk_version: 4.44.0
 app_file: app.py
 pinned: false
-tags:
-- anycoder
 ---
 
 # AI Chatbot with Hugging Face Model
@@ -20,6 +18,15 @@ This is a simple chatbot application built with Gradio and powered by a Hugging
 - Conversational AI using Microsoft's DialoGPT model
 - Gradio interface for easy interaction
 - Maintains conversation history
+- Supports Hugging Face token for accessing private models or increased rate limits
+
+## Setup
+
+To use a Hugging Face token (optional but recommended for better performance):
+
+1. Create a Hugging Face account at https://huggingface.co
+2. Generate a token at https://huggingface.co/settings/tokens
+3. In your Space settings, add the token as an environment variable named `HF_TOKEN`
 
 ## Usage
 

app.py

@@ -2,10 +2,12 @@ import gradio as gr
 from transformers import pipeline, Conversation
 import os
 
-# Load the conversational model
+# Load the conversational model with HF token support
 # Using DialoGPT-small for a simple chatbot
 # You can change this to any other conversational model like 'microsoft/DialoGPT-medium' or 'facebook/blenderbot-400M-distil-3B'
-chatbot_model = pipeline("conversational", model="microsoft/DialoGPT-small")
+# The app will use the HF_TOKEN environment variable if set
+token = os.getenv('HF_TOKEN')
+chatbot_model = pipeline("conversational", model="microsoft/DialoGPT-small", token=token)
 
 def chat(message, history):
     # Convert history to Conversation format
@@ -31,7 +33,7 @@ def chat(message, history):
 demo = gr.ChatInterface(
     fn=chat,
     title="AI Chatbot",
-    description="Chat with an AI powered by Hugging Face transformers.",
+    description="Chat with an AI powered by Hugging Face transformers. <a href='https://huggingface.co/spaces/akhaliq/anycoder' target='_blank'>Built with anycoder</a>",
     theme=gr.themes.Soft()
 )
 

requirements.txt

@@ -2,4 +2,5 @@ gradio
 transformers
 torch
 tokenizers
-accelerate
+accelerate
+huggingface_hub