Spaces:
Build error
Build error
Update app.py
Browse files
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,
|
| 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")
|