ImpactInsights commited on
Commit
c524b47
1 Parent(s): d2e8961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -2,6 +2,11 @@ import streamlit as st
2
  from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
3
  import torch
4
 
 
 
 
 
 
5
  #Sidebar menu
6
  st.sidebar.title('Menu')
7
 
@@ -36,7 +41,7 @@ if chatbot:
36
  if text:
37
  input_ids = tokenizer(text, return_tensors="pt").to("cuda")
38
  outputs = model.generate(**input_ids)
39
- st.write(print(tokenizer.decode(outputs[0])))
40
 
41
 
42
 
 
2
  from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
3
  import torch
4
 
5
+ from huggingface_hub import HfApi
6
+
7
+
8
+
9
+
10
  #Sidebar menu
11
  st.sidebar.title('Menu')
12
 
 
41
  if text:
42
  input_ids = tokenizer(text, return_tensors="pt").to("cuda")
43
  outputs = model.generate(**input_ids)
44
+ st.write(tokenizer.decode(outputs[0]))
45
 
46
 
47