trying stuff

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "EleutherAI/gpt-neo-125M"
+model_name = "flax-community/gpt-neo-125M-apps"
 # define model and tokenizer
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -9,8 +9,17 @@ 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)
+    start = len(input_ids[0])
+    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()
+inputs = [
+    gr.inputs.Textbox(placeholder="Define a problem here ...", lines=5)
+]
+
+gr.Interface(
+    generate_solution, 
+    inputs=inputs, 
+    outputs="text",
+    title="Coding problem solver",
+).launch()