adeel300 commited on
Commit
099ae28
·
verified ·
1 Parent(s): c92c79e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ # Load the Hugging Face model
5
+ model_name = "adeel300/QA_T5_small"
6
+ nlp = pipeline("text-generation", model=model_name)
7
+
8
+ # Streamlit app
9
+ st.title("Hugging Face Model Inference")
10
+
11
+ st.write("Enter your text below and get the model's response.")
12
+
13
+ # Input text
14
+ input_text = st.text_area("Input Text", value="", height=200)
15
+
16
+ if st.button("Generate Response"):
17
+ if input_text:
18
+ with st.spinner("Generating response..."):
19
+ response = nlp(input_text)
20
+ st.success("Response generated!")
21
+ st.text_area("Response", value=response[0]['generated_text'], height=200)
22
+ else:
23
+ st.warning("Please enter some text to generate a response.")