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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
  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="HuggingFaceTB/SmolLM2-1.7B") # General conversational model
8
 
9
  def generate_conversation(topic, turns):
 
3
  import torch
4
 
5
  # Load the physician and patient models via Hugging Face Model Hub
6
+ physician = pipeline("text-generation", model="HuggingFaceTB/SmolLM2-1.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):