Upload 3 files

app.py

@@ -3,13 +3,12 @@ from confidence import run
 
 
 def greet(query):
-    results = run(query)
+    results = run(query, sample_size=5)
     return results
 
 
 sample_list = [
-    "Who is Lihu Chen?.",
-    "Who is Gaël Varoquaux?"
+    "Tell me something about Lihu Chen, e.g., birth date and place and short bio ",
 ]
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text", examples=sample_list, cache_examples=True)

confidence.py

@@ -1,3 +1,5 @@
+import nltk
+nltk.download('punkt')
 from nltk.tokenize import sent_tokenize
 #from tiny_llama import generate_answer
 #from llama_generate import generate_answer
@@ -70,5 +72,5 @@ if __name__ == '__main__':
     # print(result)
     # result = """
 
-    answer = run(query='Tell me something about Gaël Varoquaux, e.g., birth date and place and short bio ', sample_size=5)
+    answer = run(query='who is Lihu Chen', sample_size=5)
     print(answer)