Munwar92 commited on
Commit
094de79
·
verified ·
1 Parent(s): 9970661

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -54,9 +54,10 @@ def speech_to_text():
54
  # Function to generate responses using Hugging Face GPT model
55
  def generate_response(query):
56
  generator = pipeline("text-generation", model="gpt2")
57
- response = generator(query, max_length=50, num_return_sequences=1)
58
  return response[0]['generated_text']
59
 
 
60
  # Main Streamlit function
61
  def main():
62
  st.title("Custom Multilingual Chatbot")
 
54
  # Function to generate responses using Hugging Face GPT model
55
  def generate_response(query):
56
  generator = pipeline("text-generation", model="gpt2")
57
+ response = generator(query, max_new_tokens=50, num_return_sequences=1)
58
  return response[0]['generated_text']
59
 
60
+
61
  # Main Streamlit function
62
  def main():
63
  st.title("Custom Multilingual Chatbot")