Deploy Spreadsheet-RL 4B inference Space

README.md

@@ -1,13 +1,17 @@
 ---
-title: Spreadsheet Rl 4b
-emoji: 🌍
-colorFrom: indigo
-colorTo: purple
+title: Spreadsheet RL 4B
+emoji: 📊
+colorFrom: green
+colorTo: blue
 sdk: gradio
 sdk_version: 6.24.0
-python_version: '3.13'
 app_file: app.py
 pinned: false
+license: apache-2.0
+models:
+  - Spreadsheet-RL/Spreadsheet-RL-4B
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Spreadsheet RL 4B
+
+Remote text generation with `Spreadsheet-RL/Spreadsheet-RL-4B`.

app.py

@@ -0,0 +1,14 @@
+import gradio as gr
+from transformers import pipeline
+
+pipe = pipeline("text-generation", model="Spreadsheet-RL/Spreadsheet-RL-4B")
+
+
+def generate(prompt):
+    messages = [
+        {"role": "user", "content": prompt},
+    ]
+    return pipe(messages)[0]["generated_text"][-1]["content"]
+
+
+gr.Interface(fn=generate, inputs="text", outputs="text").launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers==5.15.0
+torch==2.13.0