Spaces:
Sleeping
Sleeping
ImpactInsights
commited on
Commit
•
c524b47
1
Parent(s):
d2e8961
Update app.py
Browse files
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(
|
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 |
|