Updating req.txt

README.md

@@ -8,7 +8,7 @@ sdk_version: 1.32.2
 app_file: app.py
 pinned: false
 license: mit
-short_description: Century classification for classical Chinese texts
+short_description: Century classification for classical Korean/Chinese texts
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,12 +1,23 @@
+"""
+HuggingFace Spaces that:
+- loads in HanmunRoBERTa model https://huggingface.co/bdsl/HanmunRoBERTa
+- optionally strips text of punctuation and unwanted charactesr
+- predicts century for the input text
+- Visualizes prediction scores for each century
+
 # https://huggingface.co/blog/streamlit-spaces
 # https://huggingface.co/docs/hub/en/spaces-sdks-streamlit
 
+"""
+
 import streamlit as st
 from transformers import pipeline
 from string import punctuation
 import pandas as pd
+from huggingface_hub import InferenceClient
+client = InferenceClient(model="bdsl/HanmunRoBERTa")
 
-# Load the pipeline with your specific model
+# Load the pipeline with the HanmunRoBERTa model
 model_pipeline = pipeline(task="text-classification", model="bdsl/HanmunRoBERTa")
 
 # Streamlit app layout
@@ -20,7 +31,6 @@ remove_punct = st.checkbox(label="Remove punctuation", value=True)
 # Text area for user input
 input_str = st.text_area("Input text", height=275)
 
-
 # Remove punctuation if checkbox is selected
 if remove_punct and input_str:
 	# Specify the characters to remove

requirements.txt

@@ -1,4 +1,3 @@
 pandas==2.0.3
-transformers==4.39.0
-tensorflow
-torch
+transformers
+tensorflow