Add package argilla manually override within code

app.py

@@ -1,4 +1,13 @@
 #!/usr/bin/env python
+if True:
+    import subprocess
+    import sys
+
+    def install_package(package_name):
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
+
+    # Example usage:
+    install_package("argilla==2.0.0rc1")
 
 import os
 from threading import Thread
@@ -10,14 +19,14 @@ import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
-from .chat_interface_preference import ChatInterface
+from chat_interface_preference import ChatInterface
 
 MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024
 MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "8192"))
 
 if torch.cuda.is_available():
-    model_id = "davidberenstein1957/ultra-feedback-dutch-cleaned-hq-spin-geitje-7b-ultra-sft_iter2"
+    model_id = "Qwen/Qwen2-0.5B-Instruct"
     model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 style = "<style>.user-message,.system-message{display:flex;margin:10px}.user-message .message-content{background-color:#c2e3f7;color:#000}.system-message .message-content{background-color:#f5f5f5;color:#000}.message-content{padding:10px;border-radius:10px;max-width:70%;word-wrap:break-word}.container{display:flex;justify-content:space-between}.column{width:48%}</style>"
@@ -87,9 +96,8 @@ chat_interface = ChatInterface(
     chatbot=gr.Chatbot(
         height=450, label="GEITje-SPIN", show_share_button=True, avatar_images=(None, "geitje-logo.jpg")
     ),
-    style=style,
+    css=style,
     rg_dataset=dataset,
-    # textbox=gr.Textbox(value="Typ een bericht…"),
     cache_examples=False,
     additional_inputs=[
         gr.Slider(
@@ -159,11 +167,6 @@ Vat bovenstaand artikel samen"""
     description="""\
     <a href="https://huggingface.co/davidberenstein1957/ultra-feedback-dutch-cleaned-hq-spin-geitje-7b-ultra-sft_iter2">GEITje 7B SPIN iter 2</a> is een geavanceerde versie van GEITje, verder getraind op uitgebreide chat datasets en ook op een preferentiedataset om beter te aligneren met het gedrag van een gewenste chatbot op basis van het [SPIN algoritme en code](https://github.com/argilla-io/distilabel-spin-dibt/), in dit geval gpt-4-turbo. De data is net iets anders dan de originele dataset want we hebben de schoonmaak voor UltraFeedback gebruikt van Argilla, de uiteindelijke dataset is [hier](https://huggingface.co/datasets/BramVanroy/ultra_feedback_dutch_cleaned) te vinden.
     """,
-    submit_btn="Genereer",
-    stop_btn="Stop",
-    retry_btn="🔄 Opnieuw",
-    undo_btn="↩️ Ongedaan maken",
-    clear_btn="🗑️ Wissen",
 )
 
 with gr.Blocks(css="style.css") as demo:

requirements.txt

@@ -6,4 +6,3 @@ sentencepiece==0.2.0
 spaces==0.28.3
 torch==2.0.1
 transformers==4.41.2
-argilla==2.0.0rc1 --install-option="--no-deps"