charliebaby2023 commited on
Commit
72a8b28
·
verified ·
1 Parent(s): b385c52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -16
app.py CHANGED
@@ -53,29 +53,18 @@ executor = ThreadPoolExecutor(max_workers=num_models)
53
 
54
 
55
  def gen_fn(model_str, prompt):
56
- global model_name,model_scores, processed_models_count, index_tracker # Declare global variables
57
 
58
 
59
 
60
 
61
  if model_str == 'NA':
62
  return None
63
- model_name = None # Initialize `model_name` to avoid reference errors
64
- current_model_name = None # Initialize `current_model_name` to avoid reference errors
65
- current_model_index = None # Initialize `current_model_index` to avoid reference errors
66
 
67
-
68
- def generate_image():
69
- global index_tracker
70
- """Inner function to generate the image."""
71
- index_tracker = (index_tracker + 1) % len(models)
72
- current_model_index = index_tracker
73
- current_model_name = models[current_model_index]
74
- combined_prompt = f"{prompt} {randint(0, 9999)}"
75
- response = models_load[model_str](f'{combined_prompt}')
76
- return model_name, response
77
-
78
- try:
79
  index_tracker = (index_tracker + 1) % len(models)
80
  current_model_index = index_tracker
81
  current_model_name = models[current_model_index]
@@ -119,7 +108,10 @@ def gen_fn(model_str, prompt):
119
  return None
120
 
121
 
 
 
122
 
 
123
 
124
 
125
  def make_me():
 
53
 
54
 
55
  def gen_fn(model_str, prompt):
56
+ global model_scores, processed_models_count, index_tracker # Declare global variables
57
 
58
 
59
 
60
 
61
  if model_str == 'NA':
62
  return None
63
+ # model_name = None # Initialize `model_name` to avoid reference errors
64
+ # current_model_name = None # Initialize `current_model_name` to avoid reference errors
65
+ # current_model_index = None # Initialize `current_model_index` to avoid reference errors
66
 
67
+ try:
 
 
 
 
 
 
 
 
 
 
 
68
  index_tracker = (index_tracker + 1) % len(models)
69
  current_model_index = index_tracker
70
  current_model_name = models[current_model_index]
 
108
  return None
109
 
110
 
111
+
112
+
113
 
114
+
115
 
116
 
117
  def make_me():