Update app.py

app.py

@@ -24,12 +24,21 @@ time_load_model_end = time.time()
 elapsed_time = time_load_model_end - time_load_model_start
 print(f"Elapsed time to load the model: {elapsed_time:.2f} sec")
 
-
-def get_answer(llm):
-
-    return "tekst output"
-
-
-#gr.ChatInterface(get_llama_response).launch()
+def get_answer(chatbot, input_text):
+    start_time = time.time()
+    print(f"Processing the input\n {input_text}\n")
+    print('Processing the answer....')
+    conversation = Conversation(input_text)
+    print(f"Conversation(input_text): {conversation}")
+    output = (chatbot(conversation))[1]['content']
+    elapsed_time = time.time() - start_time
+    #Add the last print statement to the output variable
+    output += f"\nAnswered in {elapsed_time:.1f} seconds, Nr generated words: {count_words(output)}"
+    
+    return output
+
+
+
+#gr.ChatInterface(get_answer(llm, text)).launch()
 demo = gr.Interface(fn=get_answer, inputs="text", outputs="text")
 demo.launch(share=True)