Update app.py

knn model file

app.py

@@ -13,7 +13,7 @@ from sklearn.feature_extraction.text import CountVectorizer
 import joblib
 
 # Load the saved model
-model_file = "drug_finder_big.joblib"
+model_file = "drug_finder_big_knn.joblib"
 loaded_model = load(model_file)
 
 def predict_drug(text_input):
@@ -24,7 +24,7 @@ def predict_drug(text_input):
     drug_dosage = prediction[1]
     #drug_side_effects = prediction[3]
 
-    output_text = f"USES:\n {drug_uses} \n\nDOSAGE:\n {drug_dosage} \n"
+    output_text = f"USES:\n\n {drug_uses} \n\nDOSAGE:\n\n {drug_dosage} \n"
     return output_text
 
 # Create the interface