Update app.py

app.py

@@ -1,4 +1,15 @@
 import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "hacer201145/Hasex0.1-355M"
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+def generate_response(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    output = model.generate(inputs["input_ids"], max_length=150)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
 with gr.Blocks() as demo:
-    gr.Interface.load("huggingface", "hacer201145/Hasex0.1-355M")
-demo.launch()
+    with gr.Row():
+        model_input = gr.Textbox(label="Enter your prompt", placeholder="Type something...")
+        model_output = gr.Textbox(label="Generated response")
+    model_input.submit(generate_response, inputs=model_input, outputs=model_output)
+demo.launch()