Adding requirements.txt

app.py

@@ -2,85 +2,26 @@ from collections.abc import Sequence
 import random
 
 import gradio as gr
-import immutabledict
-import spaces
-import torch
-
-
-#### Version 1: Baseline
-# Step 1: Select and load your model
-# Step 2: Load the test dataset (4-5 examples)
-# Step 3: Run generation with and wihtout watermarking, display the outputs
-# Step 4: User clicks the reveal button to see the watermarked vs not gens
-
-#### Version 2: Gamification
-# Stesp 1-3 the same
-# Step 4: User marks specific generations as watermarked
-# Step 5: User clicks the reveal button to see the watermarked vs not gens
 
 # If the watewrmark is not detected, consider the use case. Could be because of
 # the nature of the task (e.g., fatcual responses are lower entropy) or it could
 # be another
 
-GEMMA_2B = 'google/gemma-2b'
+_GEMMA_2B = 'google/gemma-2b'
 
-PROMPTS: tuple[str] = (
+_PROMPTS: tuple[str] = (
     'prompt 1',
     'prompt 2',
     'prompt 3',
     'prompt 4',
 )
 
-WATERMARKING_CONFIG = immutabledict.immutabledict({
-    "ngram_len": 5,
-    "keys": [
-        654,
-        400,
-        836,
-        123,
-        340,
-        443,
-        597,
-        160,
-        57,
-        29,
-        590,
-        639,
-        13,
-        715,
-        468,
-        990,
-        966,
-        226,
-        324,
-        585,
-        118,
-        504,
-        421,
-        521,
-        129,
-        669,
-        732,
-        225,
-        90,
-        960,
-    ],
-    "sampling_table_size": 2**16,
-    "sampling_table_seed": 0,
-    "context_history_size": 1024,
-    "device": (
-        torch.device("cuda:0")
-        if torch.cuda.is_available()
-        else torch.device("cpu")
-    ),
-})
-
 _CORRECT_ANSWERS: dict[str, bool] = {}
 
 with gr.Blocks() as demo:
   prompt_inputs = [
       gr.Textbox(value=prompt, lines=4, label='Prompt')
-      for prompt in PROMPTS
+      for prompt in _PROMPTS
   ]
   generate_btn = gr.Button('Generate')
 

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+spaces
+transformers @ git+https://github.com/sumedhghaisas2/transformers_private@synthid_text