update model count and sample func

app.py

@@ -63,13 +63,16 @@ def sample_prompt(lora_model):
         try:
             with open(file_path, 'r') as file:
                 prompts = file.readlines()
-                total_lines = len(prompts)
-                if total_lines > 0:
-                    weights = [1/(i+1) for i in range(total_lines)]
-                    sum_weights = sum(weights)
-                    normalized_weights = [w/sum_weights for w in weights]
-                    return random.choices(prompts, weights=normalized_weights, k=1)[0].strip()
-            return "No prompts found in file."
+                if lora_model=='GenEval':
+                    total_lines = len(prompts)
+                    if total_lines > 0:
+                        weights = [1/(i+1) for i in range(total_lines)]
+                        sum_weights = sum(weights)
+                        normalized_weights = [w/sum_weights for w in weights]
+                        return random.choices(prompts, weights=normalized_weights, k=1)[0].strip()
+                    return "No prompts found in file."
+                else:
+                    return random.choice(prompts).strip()
         except FileNotFoundError:
             return "Prompt file not found."
     return ""

model_call_counter.txt

@@ -1 +1 @@
-186
+190