Spaces:

Shubu0103
/

shubhz-ai-folio-chat-model

Sleeping

Shubu0103 commited on Jun 11, 2023

Commit

3466a8f

•

1 Parent(s): 07262b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,4 +18,4 @@ if text:
     features = tokenizer([input_text], return_tensors='pt')
     out = model.generate(input_ids=features['input_ids'].to(device), attention_mask=features['attention_mask'].to(device))
     if tokenizer.decode(out[0]):
-        st.json(tokenizer.decode(out[0]))

     features = tokenizer([input_text], return_tensors='pt')
     out = model.generate(input_ids=features['input_ids'].to(device), attention_mask=features['attention_mask'].to(device))
     if tokenizer.decode(out[0]):
+        st.write(tokenizer.decode(out[0]))