simple test app

README.md

@@ -2,7 +2,7 @@
 title: Code Clippy Problem Solver
 emoji: 💻
 colorFrom: blue
-colorTo: pink
+colorTo: green
 sdk: gradio
 app_file: app.py
 pinned: false

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model_name = "EleutherAI/gpt-neo-125M"
+# define model and tokenizer
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer.pad_token = tokenizer.eos_token
+
+def generate_solution(prompt, **kwargs):
+    input_ids = tokenizer(prompt, return_tensors="pt").input_ids
+    start = len(input_ids)
+    output = model.generate(input_ids, pad_token_id=tokenizer.pad_token_id, *kwargs)
+    return tokenizer.decode(output[0][start:])
+
+gr.Interface(generate_solution, inputs="text", outputs="text").launch()

requirements.txt

@@ -0,0 +1 @@
+transformers