Upload 4 files

README.md

@@ -1,19 +1,3 @@
----
-title: BudtenderGuide
-emoji: 🚀
-colorFrom: red
-colorTo: red
-sdk: docker
-app_port: 8501
-tags:
-- streamlit
-pinned: false
-short_description: Streamlit template space
----
+# Budtender AI Assistant
 
-# Welcome to Streamlit!
-
-Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
-
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+Una app que interpreta resultados de análisis de flores de cannabis y ofrece recomendaciones basadas en terpenos y cannabinoides usando un modelo LLM ligero.

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 from utils.interpret_lab_results import analyze_lab_results
 
-st.set_page_config(page_title="Cannabis Lab Result Interpreter", layout="wide")
+st.set_page_config(page_title="Budtender AI Assistant", layout="wide")
 
-st.title("Cannabis Lab Result Analyzer")
+st.title("Budtender AI Assistant")
 
 st.markdown("Ingresa los resultados del laboratorio con niveles de terpenos y cannabinoides.")
 

utils/interpret_lab_results.py

@@ -1,6 +1,6 @@
 from transformers import pipeline
 
-generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
+generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
 
 def analyze_lab_results(input_text):
     prompt = (
@@ -10,5 +10,5 @@ def analyze_lab_results(input_text):
         f"{input_text}\n\n"
         "Análisis:"
     )
-    result = generator(prompt, max_length=500, do_sample=True)
+    result = generator(prompt, max_length=300, do_sample=True)
     return result[0]['generated_text'].split("Análisis:")[-1].strip()