aizanlabs commited on
Commit
ad7cee3
·
verified ·
1 Parent(s): 64f4a33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -71,8 +71,8 @@ class DocumentRetrievalAndGeneration:
71
  return generate_text
72
  def initialize_llm2(self,model_id):
73
 
74
- tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-1b3")#, local_files_only=True)
75
- model = BloomForCausalLM.from_pretrained("bigscience/bloom-1b3")#, local_files_only=True)
76
  result_length = 200
77
  inputs = tokenizer(prompt, return_tensors="pt")
78
  # return generate_text
@@ -186,9 +186,9 @@ if __name__ == "__main__":
186
  font-weight: bold; /* Make text bold */
187
  }
188
  """
189
- EXAMPLES = ["On which devices can the VIP and CSI2 modules operate simultaneously? ",
190
- "I'm using Code Composer Studio 5.4.0.00091 and enabled FPv4SPD16 floating point support for CortexM4 in TDA2. However, after building the project, the .asm file shows --float_support=vfplib instead of FPv4SPD16. Why is this happening?",
191
- "Could you clarify the maximum number of cameras that can be connected simultaneously to the video input ports on the TDA2x SoC, considering it supports up to 10 multiplexed input ports and includes 3 dedicated video input modules?"]
192
 
193
  interface = gr.Interface(
194
  fn=doc_retrieval_gen.qa_infer_gradio,
 
71
  return generate_text
72
  def initialize_llm2(self,model_id):
73
 
74
+ tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-3b")#, local_files_only=True)
75
+ model = BloomForCausalLM.from_pretrained("bigscience/bloom-3b")#, local_files_only=True)
76
  result_length = 200
77
  inputs = tokenizer(prompt, return_tensors="pt")
78
  # return generate_text
 
186
  font-weight: bold; /* Make text bold */
187
  }
188
  """
189
+ EXAMPLES = ["What are the main types of blood cancer, and how do they differ in terms of symptoms, progression, and treatment options? ",
190
+ "What are the latest advancements in the treatment of blood cancer, and how do they improve patient outcomes compared to traditional therapies?",
191
+ "How do genetic factors and environmental exposures contribute to the risk of developing blood cancer, and what preventive measures can be taken?"]
192
 
193
  interface = gr.Interface(
194
  fn=doc_retrieval_gen.qa_infer_gradio,