Spaces:
Sleeping
Sleeping
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 |
|