test mistral and flux 2

app.py

@@ -19,13 +19,14 @@ MAX_IMAGE_SIZE = 2048
 
 
 def refine_prompt(prompt):
+    gr.Info("Starting process")
     chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
     messages = [
         {"role": "system", "content": "You are a product designer. You will get a basic prompt of product request and you need to imagine a new product design to satisfy that need. Produce an extended description of product front view that will be use by Flux to generate a visual"},
-        {"role": "user", "content": "a castle schoolbag"},
+        {"role": "user", "content": prompt},
     ]
-    chatbot(messages)
-    refined_prompt = chatbot(prompt)
+    refined_prompt = chatbot(messages)
+    gr.Info("Ending process")
     return refined_prompt
 
 @spaces.GPU()

requirements.txt

@@ -29,4 +29,5 @@ https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffras
 accelerate
 git+https://github.com/huggingface/diffusers.git
 invisible_watermark
-sentencepiece
+sentencepiece
+protobuf