Update app.py

app.py

@@ -10,8 +10,7 @@ import torch
 #    model = AutoModelForCausalLM.from_pretrained("facebook/opt-1.3b", torch_dtype=torch.float16).cuda()
 #    tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b", use_fast=False)
 #    return model, tokenizer
-st.title('OPT-1.3 Bilion parameter')
-st.subtitle('Text-generation example')
+st.markdown('## OPT-1.3 Bilion parameter')
 
 with st.spinner('Loading OPT-1.3b Model...'):
     generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)