Update app.py from anycoder

app.py

@@ -1,19 +1,71 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import InferenceClient
 
-tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct")
-model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM2-1.7B-Instruct")
+# Initialize the Hugging Face Inference client
+client = InferenceClient()
 
 def generate_text(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_length=200)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    try:
+        # Generate text using the Inference API
+        generated_text = client.text_generation(
+            prompt,
+            model="HuggingFaceTB/SmolLM2-1.7B-Instruct",
+            max_new_tokens=200,
+            do_sample=True,
+            temperature=0.7
+        )
+        return generated_text
+    except Exception as e:
+        return f"Error generating text: {str(e)}"
 
+# Create the Gradio interface with modern theme
 with gr.Blocks() as demo:
+    gr.Markdown("# SmolLM Text Generation 🚀")
+    gr.Markdown("Built with [anycoder](https://huggingface.co/spaces/akhaliq/anycoder)")
+    
     with gr.Row():
-        input_text = gr.Textbox(label="Input Prompt")
-        output_text = gr.Textbox(label="Generated Text")
-    generate_btn = gr.Button("Generate")
-    generate_btn.click(generate_text, inputs=input_text, outputs=output_text)
+        with gr.Column():
+            input_text = gr.Textbox(
+                label="Input Prompt",
+                placeholder="Enter your prompt here...",
+                lines=5
+            )
+            generate_btn = gr.Button("Generate", variant="primary")
+        
+        with gr.Column():
+            output_text = gr.Textbox(
+                label="Generated Text",
+                interactive=False,
+                lines=10
+            )
+    
+    # Add examples
+    examples = [
+        ["Explain quantum computing in simple terms"],
+        ["Write a short story about a robot learning to love"],
+        ["How do I make a perfect omelette?"]
+    ]
+    
+    gr.Examples(
+        examples=examples,
+        inputs=input_text,
+        outputs=output_text,
+        fn=generate_text,
+        cache_examples=True
+    )
+    
+    generate_btn.click(
+        fn=generate_text,
+        inputs=input_text,
+        outputs=output_text,
+        api_visibility="public"
+    )
 
-demo.launch()
+# Launch with modern theme and settings
+demo.launch(
+    theme=gr.themes.Soft(primary_hue="blue"),
+    footer_links=[
+        {"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"},
+        {"label": "Model", "url": "https://huggingface.co/HuggingFaceTB/SmolLM2-1.7B-Instruct"}
+    ]
+)