to(cuda)

lora.py

@@ -34,7 +34,7 @@ def add_ingredient(ingredient, quantity):
 
 # Function to enable/disable add button
 def validate_inputs(ingredient, quantity):
-    if ingredient and quantity > 0:
+    if ingredient and int(quantity) > 0:
         return gr.update(interactive=True)
     return gr.update(interactive=False)
 
@@ -62,7 +62,7 @@ def submit_to_model():
         tokenize=True,
         add_generation_prompt=True,  # Must add for generation
         return_tensors="pt",
-    )
+    ).to("cuda")
 
     text_streamer = TextIteratorStreamer(tokenizer, skip_prompt=True)