ari7cr commited on
Commit
7f9e70c
·
verified ·
1 Parent(s): c36c4ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  # Load the physician and patient models via Hugging Face Model Hub
6
  physician = pipeline("text-generation", model="BioMistral/BioMistral-7B") # Replace with actual medical model
7
- patient = pipeline("text-generation", model="BioMistral/BioMistral-7B") # General conversational model
8
 
9
  def generate_conversation(topic, turns):
10
  conversation = []
 
4
 
5
  # Load the physician and patient models via Hugging Face Model Hub
6
  physician = pipeline("text-generation", model="BioMistral/BioMistral-7B") # Replace with actual medical model
7
+ patient = pipeline("text-generation", model="HuggingFaceTB/SmolLM2-1.7B") # General conversational model
8
 
9
  def generate_conversation(topic, turns):
10
  conversation = []