import os
from huggingface_hub import login
from transformers import pipeline
import gradio as gr

# 🔐 Login using the secret token (set in Space Settings > Secrets)
login(os.getenv("HF_TOKEN"))

# 🚀 Load the gated model
pipe = pipeline("text-generation", model="google/gemma-2b-it")

# 🧠 Generation function
def generate(prompt):
    output = pipe(prompt, max_new_tokens=100)
    return output[0]["generated_text"]

# 🎛️ Gradio Interface
iface = gr.Interface(fn=generate, inputs="text", outputs="text")
iface.launch()