Update app.py

app.py

@@ -1,18 +1,25 @@
 import time
 import gradio as gr
 from ctransformers import AutoModelForCausalLM
+from spellchecker import SpellChecker
 
-# Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
 llm = AutoModelForCausalLM.from_pretrained("TheBloke/WizardLM-7B-uncensored-GGUF", model_file="WizardLM-7B-uncensored.Q3_K_M.gguf", model_type="llama", stream=True)
 history = [""]
+spell = SpellChecker()
+
+def correct_words(text):
+    words = text.split()
+    corrected_words = [spell.correction(word) for word in words]
+    corrected_text = ' '.join(corrected_words)
+    return corrected_text
 
 def generate_response(message):
     global history
     for text in llm(message):
-        
         response = ''.join(text)
         time.sleep(2)
-        history.append(response)
+        corrected_response = correct_words(response)
+        history.append(corrected_response)
         yield ' '.join(history)
     # Clear the history list after the last response
     history = ["Chatbot:"]
@@ -24,4 +31,4 @@ def chatbot(message, history):
         yield response
 
 iface = gr.ChatInterface(chatbot)
-iface.launch()
+iface.launch()