Update app.py

app.py

@@ -5,14 +5,18 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("qbwmwsap/amber-model-mine")
 model = AutoModelForCausalLM.from_pretrained("qbwmwsap/amber-model-mine")
 
-text = st.text_area("enter some text!")
+st.title('Amber-Model-Mine')
+
+text = st.text_input('Enter some text')
 
 if text:
+    # Preprocess your data here
     input_ids = tokenizer(text, return_tensors="pt").input_ids
-    
+
     # Run the model and get the output
     outputs = model.generate(input_ids)
-    
+
     # Postprocess your output here if necessary    
     output = tokenizer.decode(outputs[0])
-    st.json(output)
+
+    st.write(output)