Chan-Y commited on
Commit
76b375c
1 Parent(s): 4519bd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ llm = HuggingFaceEndpoint(
15
  # Define the function to process user input
16
  def classify_text(text):
17
  prompt = f"""Classify the following text into a category or topic. You dont need to write specific informations or explanations, only return the categories.
18
- {text}\nCategories:"""
19
  results = llm.invoke(prompt)
20
  return results
21
 
 
15
  # Define the function to process user input
16
  def classify_text(text):
17
  prompt = f"""Classify the following text into a category or topic. You dont need to write specific informations or explanations, only return the categories.
18
+ {text}\nCategories of the text:"""
19
  results = llm.invoke(prompt)
20
  return results
21