Update app.py

app.py

@@ -10,10 +10,12 @@ 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')
 
 with st.spinner('Loading OPT-1.3b Model...'):
     generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)    
-st.success('Model loaded correctly!')
+    st.success('Model loaded correctly!')
 
 #@st.cache(allow_output_mutation=True)
 #def opt_model(prompt, model, tokenizer, num_sequences = 1, max_length = 50):  
@@ -28,7 +30,7 @@ st.success('Model loaded correctly!')
 prompt= st.text_area('Your prompt here',
  '''Hello, I'm am conscious and''') 
  
-answer = generator(prompt, max_length=100,no_repeat_ngram_size=3, early_stopping=True, num_beams=10)
+answer = generator(prompt, max_length=100, no_repeat_ngram_size=3, early_stopping=True, num_beams=10)
  
 #answer = opt_model(prompt, model, tokenizer,)
 #lst = ['ciao come stai sjfsbd dfhsdf  fuahfuf  feuhfu wefwu ']