Premchandyadav369 commited on
Commit
72bef29
·
1 Parent(s): 02dab15

fix(app): sync demo module-level instance and queue with space

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -102,7 +102,8 @@ def generate_with_telemetry(prompt: str, max_tokens: int = 64, temperature: floa
102
 
103
  for step in range(max_tokens):
104
  with torch.no_grad():
105
- outputs = MODEL(curr_ids[:, -1:], past_key_values=past_key_values, use_cache=True)
 
106
  logits = outputs.logits[:, -1, :]
107
  past_key_values = outputs.past_key_values
108
 
@@ -283,7 +284,10 @@ def build_app():
283
  return demo
284
 
285
 
 
 
 
286
  if __name__ == "__main__":
287
- app = build_app()
288
- if app:
289
- app.launch(server_name="0.0.0.0", server_port=7860, share=False)
 
102
 
103
  for step in range(max_tokens):
104
  with torch.no_grad():
105
+ model_in = curr_ids if step == 0 else curr_ids[:, -1:]
106
+ outputs = MODEL(model_in, past_key_values=past_key_values, use_cache=True)
107
  logits = outputs.logits[:, -1, :]
108
  past_key_values = outputs.past_key_values
109
 
 
284
  return demo
285
 
286
 
287
+ # Instantiate demo at module level for Hugging Face Spaces & Gradio discovery
288
+ demo = build_app()
289
+
290
  if __name__ == "__main__":
291
+ if demo is not None:
292
+ demo.queue().launch(server_name="0.0.0.0", server_port=7860)
293
+