Update app.py

app.py

@@ -1,7 +1,6 @@
 import transformers
 import re
 from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForCausalLM
-from vllm import LLM, SamplingParams
 import torch
 import gradio as gr
 import json
@@ -70,20 +69,6 @@ def split_text(text, max_tokens=500):
     return chunks
 
 
-#Curtesy of claude
-def generate_html_diff(old_text, new_text):
-    d = difflib.Differ()
-    diff = list(d.compare(old_text.split(), new_text.split()))
-    
-    html_diff = []
-    for word in diff:
-        if word.startswith('  '):
-            html_diff.append(word[2:])
-        elif word.startswith('+ '):
-            html_diff.append(f'<span style="background-color: #90EE90;">{word[2:]}</span>')
-        # We're not adding anything for words that start with '- '
-    
-    return ' '.join(html_diff)
 
 # Class to encapsulate the Falcon chatbot
 class MistralChatBot: