rishabhsetiya commited on
Commit
056863e
·
verified ·
1 Parent(s): d98c763

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def combined_generate(prompt, max_tokens):
30
  start_times = {"Fine-tuned": time.time(), "RAG": time.time()}
31
 
32
  futures = {
33
- executor.submit(generate_answer, model, tokenizer, device, prompt, max_tokens): "Fine-tuned",
34
  executor.submit(generate_answer_rag, prompt, max_tokens): "RAG",
35
  }
36
 
 
30
  start_times = {"Fine-tuned": time.time(), "RAG": time.time()}
31
 
32
  futures = {
33
+ executor.submit(fine_tuning.generate_answer, model, tokenizer, device, prompt, max_tokens): "Fine-tuned",
34
  executor.submit(generate_answer_rag, prompt, max_tokens): "RAG",
35
  }
36